Maguiiiiiiiiiiiiiiiiiiiii's picture
init
b73dc00 verified
raw
history blame contribute delete
832 Bytes
import gradio as gr
from transformers import pipeline
# Definir función para obtener la respuesta
def obtener_respuesta(contexto, pregunta):
# Cargar el modelo pre-entrenado
model_name = "deepset/roberta-base-squad2"
nlp = pipeline('question-answering', model=model_name, tokenizer=model_name)
# Obtener la respuesta predicha
res = nlp({'question': pregunta, 'context': contexto})
return res['answer']
# Crear la interfaz de Gradio
interfaz = gr.Interface(
fn=obtener_respuesta,
inputs=[
gr.Textbox(label="Contexto"),
gr.Textbox(label="Pregunta")
],
outputs=gr.Textbox(label="Respuesta"),
title="QA con Gradio",
description="Introduce el contexto y la pregunta para obtener la respuesta.",
theme="glass"
)
# Lanzar la interfaz
interfaz.launch(share=True)