import gradio as gr
from simpletransformers.question_answering import QuestionAnsweringModel, QuestionAnsweringArgs
import constants

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]

gr.Interface(fn=question_answer, inputs=["text", "text"], outputs=[gr.outputs.Textbox(label="Answer"), gr.outputs.Textbox(label="Probability")]).launch()