finalmaybe

app.py

@@ -25,7 +25,7 @@ def answer_question(context, question):
   answer_end = torch.argmax(end_scores) + 1
 
   # Get answer tokens and convert them to string
-  answer = tokenizer.convert_ids_to_tokens(inputs["input_ids"][0])
+  answer = tokenizer.convert_ids_to_tokens(inputs["input_ids"][0][answer_start:answer_end])
   answer = "".join(answer)
 
   return answer