Update app.py
Browse files
app.py
CHANGED
|
@@ -26,7 +26,7 @@ def generate_response(passage: str, question: str) -> str:
|
|
| 26 |
|
| 27 |
response = tokenizer.batch_decode(response, skip_special_tokens=True)[0]
|
| 28 |
|
| 29 |
-
response = response[len(message[0]):].strip()
|
| 30 |
|
| 31 |
return response
|
| 32 |
|
|
|
|
| 26 |
|
| 27 |
response = tokenizer.batch_decode(response, skip_special_tokens=True)[0]
|
| 28 |
|
| 29 |
+
response = response[len(message[0]):].strip().split('\n')[0]
|
| 30 |
|
| 31 |
return response
|
| 32 |
|