Spaces:
Sleeping
Sleeping
fix: skip system output
Browse files
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|>")[
|
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 |
|