Update app.py

app.py

@@ -1,16 +1,15 @@
 import gradio as gr
 from transformers import pipeline
 
-# Load RoBERTa QA model
+# Load QA Model
 qa_pipeline = pipeline(
-    "question-answering",
-    model="deepset/roberta-base-squad2",
-    tokenizer="deepset/roberta-base-squad2"
+    task="question-answering",
+    model="deepset/roberta-base-squad2"
 )
 
-# Function for answering question
+# Prediction Function
 def answer_question(context, question):
-    if context.strip() == "" or question.strip() == "":
+    if not context.strip() or not question.strip():
         return "Please enter both context and question."
 
     result = qa_pipeline(
@@ -20,14 +19,14 @@ def answer_question(context, question):
 
     return result["answer"]
 
-# Gradio Interface
-app = gr.Interface(
+# Gradio UI
+demo = gr.Interface(
     fn=answer_question,
     inputs=[
         gr.Textbox(
             lines=8,
             label="Context",
-            placeholder="Enter paragraph / context here..."
+            placeholder="Enter paragraph or context here..."
         ),
         gr.Textbox(
             lines=2,
@@ -37,7 +36,8 @@ app = gr.Interface(
     ],
     outputs=gr.Textbox(label="Answer"),
     title="RoBERTa Question Answering System",
-    description="Ask questions from given context using RoBERTa model."
+    description="Ask questions from the given context using RoBERTa."
 )
 
-app.launch()
+# Launch for Hugging Face Spaces
+demo.launch(server_name="0.0.0.0", server_port=7860)