gouravbhadraDev commited on
Commit
04f3b88
·
verified ·
1 Parent(s): effce56

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
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
- # DeepSeek models usually do not have a special reasoning delimiter, so return empty reasoning
165
- return "", generated_text.strip()
 
 
 
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):