krisha06 commited on
Commit
37d446a
·
verified ·
1 Parent(s): aee09f8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -53,8 +53,8 @@ def chat(instruction):
53
  # print("🧠 Full output:", repr(response))
54
 
55
  # Split and return the assistant's answer
56
- if "### ASSISTANT:" in response:
57
- return response.split("### ASSISTANT:")[-1].strip()
58
  else:
59
  return response.strip()
60
 
 
53
  # print("🧠 Full output:", repr(response))
54
 
55
  # Split and return the assistant's answer
56
+ if "### Answer:" in response:
57
+ return response.split("### Answer:")[-1].strip()
58
  else:
59
  return response.strip()
60