MythSus commited on
Commit
6505fc0
·
verified ·
1 Parent(s): 175b0dc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = text.split("Question:")[0].strip()
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)