Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -161,8 +161,11 @@ def generate_deepseek(prompt: str) -> (str, str):
|
|
| 161 |
)
|
| 162 |
generated_text = deepseek_tokenizer.decode(outputs[0], skip_special_tokens=True)
|
| 163 |
|
| 164 |
-
|
| 165 |
-
|
|
|
|
|
|
|
|
|
|
| 166 |
|
| 167 |
|
| 168 |
def generate_t5(prompt: str) -> (str, str):
|
|
|
|
| 161 |
)
|
| 162 |
generated_text = deepseek_tokenizer.decode(outputs[0], skip_special_tokens=True)
|
| 163 |
|
| 164 |
+
if "</think>" in generated_text:
|
| 165 |
+
reasoning_content, content = generated_text.split("</think>", 1)
|
| 166 |
+
return reasoning_content.strip(), content.strip()
|
| 167 |
+
else:
|
| 168 |
+
return "", generated_text.strip()
|
| 169 |
|
| 170 |
|
| 171 |
def generate_t5(prompt: str) -> (str, str):
|