Spaces:
Running on Zero
Running on Zero
electblake commited on
Commit ·
89354a7
1
Parent(s): 604e6eb
fix: return only the final model answer
Browse files
app.py
CHANGED
|
@@ -98,7 +98,7 @@ def generate(
|
|
| 98 |
top_p=0.95,
|
| 99 |
top_k=20,
|
| 100 |
)
|
| 101 |
-
return completion["choices"][0]["message"]["content"].strip()
|
| 102 |
|
| 103 |
|
| 104 |
CSS = """
|
|
|
|
| 98 |
top_p=0.95,
|
| 99 |
top_k=20,
|
| 100 |
)
|
| 101 |
+
return completion["choices"][0]["message"]["content"].rsplit("</think>", 1)[-1].strip()
|
| 102 |
|
| 103 |
|
| 104 |
CSS = """
|