Update app.py

app.py

@@ -31,9 +31,12 @@ def summarize_document(document):
 # Function to get answer to question
 def get_answer(question, context):
     inputs = qa_tokenizer(question, context, return_tensors="pt", padding=True, truncation=True, max_length=512)
-    start_positions, end_positions = qa_model(**inputs)
-    answer_start = torch.argmax(start_positions)
-    answer_end = torch.argmax(end_positions) + 1
+    outputs = qa_model(**inputs)
+    start_scores = outputs.start_logits
+    end_scores = outputs.end_logits
+
+    answer_start = torch.argmax(start_scores)
+    answer_end = torch.argmax(end_scores) + 1
     answer = qa_tokenizer.convert_tokens_to_string(qa_tokenizer.convert_ids_to_tokens(inputs["input_ids"][0][answer_start:answer_end]))
     return answer