SauravMaheshkar commited on
Commit
9b35e27
·
unverified ·
1 Parent(s): 721133c

fix: skip system output

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -68,7 +68,7 @@ def invoke(history: HistoryType):
68
  response = pipe(input_text, do_sample=True, top_p=0.95, max_new_tokens=1024)[0][
69
  "generated_text"
70
  ]
71
- response = response.split("<|im_start|>assistant\n")[-1].split("<|im_end|>")[0]
72
  return response
73
 
74
 
 
68
  response = pipe(input_text, do_sample=True, top_p=0.95, max_new_tokens=1024)[0][
69
  "generated_text"
70
  ]
71
+ response = response.split("<|im_start|>assistant\n")[-1].split("<|im_end|>")[1]
72
  return response
73
 
74