import gradio as gr
from transformers import pipeline

# Load your model
qa_pipeline = pipeline("question-answering", model="Austinsz-Warehouse/test2")

def answer_question(context, question):
    result = qa_pipeline(question=question, context=context)
    return result['answer']

# Gradio interface
gr.Interface(
    fn=answer_question,
    inputs=["text", "text"],
    outputs="text",
    title="Question Answering with test2",
    description="Ask questions based on a given context using Austinsz-Warehouse/test2"
).launch()