app.py

import gradio as gr
from transformers import pipeline

def get_answer(question):
    # 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"]

# Create Gradio Interface
iface = gr.Interface(
    fn=get_answer,
    inputs="text",
    outputs="text",
    title="Ask Any Question",
    description="Ask factual questions and get precise answers."
)

iface.launch()