Update app.py

app.py

@@ -1,26 +1,24 @@
 import gradio as gr
 from transformers import pipeline
 
-# Use a different LLM (GPT-Neo instead of GPT-2)
-generator = pipeline('text-generation', model='EleutherAI/gpt-neo-125M')
+# Load a Question Answering model
+qa_pipeline = pipeline("question-answering", model="deepset/roberta-base-squad2")
 
-def generate_text(prompt):
-    generated = generator(
-        prompt,
-        max_length=20,          # Limit response length
-        do_sample=False,        # Make output deterministic
-        temperature=0.1,        # Reduce randomness
-        repetition_penalty=2.0  # Prevent repeating words
-    )
-    return generated[0]['generated_text']
+def get_answer(question):
+    # Define a context for factual answers
+    context = """
+    New Delhi is the capital of India. India is a country in South Asia and is the seventh-largest country by land area.
+    """
+    answer = qa_pipeline(question=question, context=context)
+    return answer["answer"]
 
-# Create the Gradio interface
+# Create Gradio Interface
 iface = gr.Interface(
-    fn=generate_text,
+    fn=get_answer,
     inputs="text",
     outputs="text",
     title="Ask Any Question",
-    description="Ask a question and get an answer using GPT-Neo."
+    description="Ask factual questions and get precise answers."
 )
 
 iface.launch()