mohammadhakimi commited on
Commit
bf67ebb
·
1 Parent(s): 92998af

fixes on response problem

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -90,12 +90,12 @@ def chat(message, history):
90
  print(response)
91
  if isinstance(response, str):
92
  # Clean up the output to remove unnecessary text
93
- response = response.split("Answer: ")[0].split("</s")[0] # Get everything after "Answer:"
94
  response = response.strip() # Remove extra spaces or newline characters
95
  return response
96
  elif isinstance(response, dict):
97
  response = response.get("answer", "Sorry, I couldn't generate an answer.")
98
- response = response.split("Answer: ")[0].split("</s")[0]
99
  response = response.strip()
100
  return response
101
  else:
 
90
  print(response)
91
  if isinstance(response, str):
92
  # Clean up the output to remove unnecessary text
93
+ response = response.split("Answer: ")[1].split("</s")[0] # Get everything after "Answer:"
94
  response = response.strip() # Remove extra spaces or newline characters
95
  return response
96
  elif isinstance(response, dict):
97
  response = response.get("answer", "Sorry, I couldn't generate an answer.")
98
+ response = response.split("Answer: ")[1].split("</s")[0]
99
  response = response.strip()
100
  return response
101
  else: