Update app.py

app.py

@@ -1,14 +1,17 @@
 import gradio as gr
 from transformers import pipeline
 
-# Load a question-answering model instead of a text generator
-qa_pipeline = pipeline("question-answering", model="deepset/roberta-base-squad2")
-
 def get_answer(question):
-    # Define a factual context
+    # Reload the pipeline every time a query is made (fixes single-response issue)
+    qa_pipeline = pipeline("question-answering", model="deepset/roberta-base-squad2")
+
+    # Define a general knowledge context
     context = """
     Washington, D.C. is the capital of the United States of America.
+    New Delhi is the capital of India.
+    London is the capital of the United Kingdom.
     """
+    
     answer = qa_pipeline(question=question, context=context)
     return answer["answer"]