Update app.py

app.py

@@ -143,9 +143,9 @@ def main():
                input_ids = inputs["input_ids"].numpy()[0]
 
                text_tokens = tokenizer.convert_ids_to_tokens(input_ids)
-               answer_start_scores, answer_end_scores = qa_model(inputs)
-               answer_start = tf.argmax(answer_start_scores, axis=1).numpy()[0]
-               answer_end = (tf.argmax(answer_end_scores, axis=1) + 1).numpy()[0] 
+               outputs = qa_model(inputs)
+               answer_start = tf.argmax(outputs.start_logits, axis=1).numpy()[0]
+               answer_end = (tf.argmax(outputs.end_logits, axis=1) + 1).numpy()[0] 
                 
                answer = tokenizer.convert_tokens_to_string(tokenizer.convert_ids_to_tokens(input_ids[answer_start:answer_end]))