Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -266,7 +266,7 @@ def qa_system(method, question):
|
|
| 266 |
pad_token_id=tokenizer.eos_token_id)
|
| 267 |
answer = tokenizer.decode(outputs[0], skip_special_tokens=True).split("Answer:")[-1].strip()
|
| 268 |
if "Question:" in answer:
|
| 269 |
-
answer =
|
| 270 |
|
| 271 |
end_time = time.time()
|
| 272 |
response_time = round(end_time - start_time, 2)
|
|
|
|
| 266 |
pad_token_id=tokenizer.eos_token_id)
|
| 267 |
answer = tokenizer.decode(outputs[0], skip_special_tokens=True).split("Answer:")[-1].strip()
|
| 268 |
if "Question:" in answer:
|
| 269 |
+
answer = answer.split("Question:")[0].strip()
|
| 270 |
|
| 271 |
end_time = time.time()
|
| 272 |
response_time = round(end_time - start_time, 2)
|