Update app.py

app.py

@@ -1,23 +1,11 @@
 import gradio as gr
-from simpletransformers.question_answering import QuestionAnsweringModel, QuestionAnsweringArgs
 import constants
+from transformers import pipeline
+
+question_answerer = pipeline("question-answering", model=constants.MODEL_NAME)
 
-model = QuestionAnsweringModel(
-    constants.MODEL_TYPE, constants.MODEL_NAME, use_cuda=constants.USE_CUDA
-)
 def question_answer(context, question):
-    to_predict = [
-        {
-            "context": context,
-            "qas": [
-                {
-                    "question": question,
-                    "id": "0",
-                }
-            ],
-        }
-    ]
-    answers, probabilities = model.predict(to_predict)
-    return answers[0]['answer'][0],probabilities[0]['probability'][0]
+    result = question_answerer(question=question, context=context)
+    return result['answer'],result['score']
 
 gr.Interface(fn=question_answer, inputs=["text", "text"], outputs=[gr.outputs.Textbox(label="Answer"), gr.outputs.Textbox(label="Probability")]).launch()