Update app.py

app.py

@@ -1,25 +1,20 @@
 import gradio as gr
 from transformers import pipeline
 
-# Load the model and tokenizer from Hugging Face
+# Load your QA model
 qa_pipeline = pipeline('question-answering', model='MaawaKhalid/Extractive-QA-Bot')
 
-# Define the function to answer questions
-def answer_question(context, question):
-    result = qa_pipeline(question=question, context=context)
+# Define a function to handle the QA
+def qa_function(context, question):
+    result = qa_pipeline({'context': context, 'question': question})
     return result['answer']
 
-# Create a simple Gradio interface
-interface = gr.Interface(
-    fn=answer_question,
+# Create a Gradio interface
+gr.Interface(
+    fn=qa_function,
     inputs=[
-        gr.inputs.Textbox(label="Context", placeholder="Enter some text to ask questions from"),
-        gr.inputs.Textbox(label="Question", placeholder="Ask a question based on the context")
+        gr.components.Textbox(label="Context", placeholder="Enter some text to ask questions from"),
+        gr.components.Textbox(label="Question", placeholder="Enter your question")
     ],
-    outputs="text",
-    title="Extractive QA Bot",
-    description="Ask questions and get answers based on the provided context."
-)
-
-# Launch the interface
-interface.launch()
+    outputs=gr.components.Textbox(label="Answer")
+).launch()