ruslanmv commited on
Commit
83034af
1 Parent(s): 3720f41

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -30,7 +30,9 @@ def askme(symptoms, question):
30
  # Remove system messages and content
31
  # Extract and return the generated text, removing the prompt
32
  # Extract only the assistant's response
33
- answer = response_text.split('<|im_start|>assistant')[-1].strip()
 
 
34
  return answer
35
 
36
 
 
30
  # Remove system messages and content
31
  # Extract and return the generated text, removing the prompt
32
  # Extract only the assistant's response
33
+
34
+ # Extract only the assistant's response
35
+ answer = response_text.split('<|im_start|>assistant')[1].split('<|im_end|>')[0].strip()
36
  return answer
37
 
38