hsuwill000 commited on
Commit
246dff9
·
verified ·
1 Parent(s): ef9af50

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -43,7 +43,7 @@ def respond(prompt, history):
43
  response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
44
  print("Decoded response:", response)
45
  # **去除 `<think>` 及其他無用內容**
46
- response = response.replace("<think>", "").replace("</think>", "").strip()
47
  # 返回回應
48
  return response
49
 
 
43
  response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
44
  print("Decoded response:", response)
45
  # **去除 `<think>` 及其他無用內容**
46
+ response = response.replace("<think>", "**THINK**").replace("</think>", "**THINK**").strip()
47
  # 返回回應
48
  return response
49