Spaces:
Running
Running
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):
|