Update app.py

app.py

@@ -1,18 +1,18 @@
 import gradio as gr
 from transformers import pipeline
 
-# Load QA model
-qa = pipeline(
+# Load lightweight QA model
+qa_pipeline = pipeline(
     "question-answering",
-    model="deepset/roberta-base-squad2"
+    model="distilbert-base-cased-distilled-squad"
 )
 
 # Function
-def answer_question(context, question):
-    if not context.strip() or not question.strip():
-        return "Please enter both context and question."
+def answer(context, question):
+    if context.strip() == "" or question.strip() == "":
+        return "Please enter context and question."
 
-    result = qa(
+    result = qa_pipeline(
         question=question,
         context=context
     )
@@ -21,22 +21,14 @@ def answer_question(context, question):
 
 # UI
 demo = gr.Interface(
-    fn=answer_question,
+    fn=answer,
     inputs=[
-        gr.Textbox(
-            lines=8,
-            label="Context",
-            placeholder="Paste paragraph here..."
-        ),
-        gr.Textbox(
-            lines=2,
-            label="Question",
-            placeholder="Ask question..."
-        )
+        gr.Textbox(lines=8, label="Context"),
+        gr.Textbox(lines=2, label="Question")
     ],
-    outputs=gr.Textbox(label="Answer"),
-    title="🔍 Question Answering System",
-    description="Context-based QA using RoBERTa model.",
+    outputs="text",
+    title="🔍 QA System",
+    description="Ask questions from given context.",
     theme=gr.themes.Soft()
 )