Update app.py

app.py

@@ -1,14 +1,12 @@
 import gradio as gr
-from bert import QA
 
-model = QA('bert-large-uncased-whole-word-masking-finetuned-squad')
+examples = [
+    ["It's rainy today but it will stop in a few hours", "What is the weather like in a few hours?"]
+]
 
-def qa_func(context, question):
-    return model.predict(context, question)["answer"]
-
-gr.Interface(qa_func,
-    [
-        gr.inputs.Textbox(lines=7, label="Paragraph"),
-        gr.inputs.Textbox(label="Question"),
-    ],
-    gr.outputs.Textbox(label="Answer")).launch()
+gr.Interface.load("huggingface/deepset/roberta-base-squad2", 
+                  inputs=[gr.inputs.Textbox(lines=10, label="Context", placeholder="Type a sentence or paragraph here."), 
+                          gr.inputs.Textbox(lines=2, label="Question", placeholder="Ask a question based on the context.")],
+                  outputs=[gr.outputs.Textbox(label="Answer"), 
+                          gr.outputs.Label(label="Probability")],                  
+                  examples=examples).launch()