QA pipeline

app.py

@@ -1,14 +1,14 @@
 import gradio as gr
 from transformers import pipeline
 
-def greet(name):
-    return "Hello " + name + "!"
+qa_model = pipeline("question-answering",'a-ware/bart-squadv2')
 
-with gr.Blocks() as demo:
-    name = gr.Textbox(label="Enter your question here")
-    submit_btn = gr.Button("Submit")
-    submit_btn.click(fn=greet, inputs=name)
-    #output = gr.Textbox(label="Output Box")
+def fetch_answer(name):
+    return qa_model(question = question, context = context)['answer']
 
-
-demo.launch()
+demo = gr.Interface(
+    fn=fetch_answer,
+    inputs=gr.Textbox(lines=2, placeholder="Name Here..."),
+    outputs="text",
+)
+demo.launch()