add.py

print('add.py 실행 시작!')

from transformers import pipeline
import gradio as gr

qna = pipeline("question-answering")

def answer_question(question, context):
    result = qna({
        "question": question,
        "context": context
    })
    if isinstance(result, list):
        return result[0]['answer']
    return result['answer']

demo = gr.Interface(
    fn=answer_question,
    inputs=["text", "text"],
    outputs="text",
    title="Question Answering"
)
demo.launch()