Commit
·
bf67ebb
1
Parent(s):
92998af
fixes on response problem
Browse files
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: ")[
|
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: ")[
|
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:
|