mohit-sarvam commited on
Commit
f102061
·
verified ·
1 Parent(s): e172e9b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -105,10 +105,10 @@ output_text = response.choices[0].message.content
105
 
106
  if "</think>" in output_text:
107
  reasoning_content = output_text.split("</think>")[0].rstrip("\n")
108
- content = output_text.split("</think>")[-1].lstrip("\n").rstrip("</s>")
109
  else:
110
  reasoning_content = ""
111
- content = output_text.rstrip("</s>")
112
 
113
  print("reasoning content:", reasoning_content)
114
  print("content:", content)
 
105
 
106
  if "</think>" in output_text:
107
  reasoning_content = output_text.split("</think>")[0].rstrip("\n")
108
+ content = output_text.split("</think>")[-1].lstrip("\n")
109
  else:
110
  reasoning_content = ""
111
+ content = output_text
112
 
113
  print("reasoning content:", reasoning_content)
114
  print("content:", content)