Spaces:
Runtime error
Runtime error
Delete app.py
Browse files
app.py
DELETED
@@ -1,58 +0,0 @@
|
|
1 |
-
# -*- coding: utf-8 -*-
|
2 |
-
"""Copia de app.ipynb
|
3 |
-
|
4 |
-
Automatically generated by Colab.
|
5 |
-
|
6 |
-
Original file is located at
|
7 |
-
https://colab.research.google.com/drive/1CWlm7_FskC5LwZaYL-sgzASC5LDAzZ15
|
8 |
-
"""
|
9 |
-
|
10 |
-
import gradio as gr
|
11 |
-
|
12 |
-
import transformers
|
13 |
-
from transformers import pipeline
|
14 |
-
import pandas as pd
|
15 |
-
|
16 |
-
|
17 |
-
from transformers import AutoModelForQuestionAnswering, AutoTokenizer
|
18 |
-
|
19 |
-
model_name = "mrm8488/distill-bert-base-spanish-wwm-cased-finetuned-spa-squad2-es"
|
20 |
-
|
21 |
-
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
22 |
-
model = AutoModelForQuestionAnswering.from_pretrained(model_name)
|
23 |
-
|
24 |
-
url= 'https://docs.google.com/spreadsheets/d/e/2PACX-1vSWJvS1tP7pp_GpF1fb-KXDyQoACbxKZlUfB4Gd3SFHnX6Jz3xs9-9t0yMOnfPd59Up50Fp8OS02epO/pub?gid=1955047855&single=true&output=csv'
|
25 |
-
df = pd.read_csv(url)
|
26 |
-
|
27 |
-
df['contexto'] = df['Descripci贸n'] + '' + df['Atributos'] + ''
|
28 |
-
|
29 |
-
contestador = pipeline('question-answering', model='mrm8488/distill-bert-base-spanish-wwm-cased-finetuned-spa-squad2-es', from_pt=True)
|
30 |
-
def obtener_origen(personaje):
|
31 |
-
columna = df[df['Personaje'] == personaje]
|
32 |
-
if columna.empty:
|
33 |
-
return f'El personaje "{personaje}" no se encuentra en la base de datos.'
|
34 |
-
origen = columna.iloc[0]['Origen']
|
35 |
-
return f'El origen de este ser mitol贸gico "{personaje}" es "{origen}".'
|
36 |
-
|
37 |
-
def responder_pregunta(personaje, pregunta):
|
38 |
-
columna = df[df['Personaje'] == personaje]
|
39 |
-
if columna.empty:
|
40 |
-
return f'El personaje "{personaje}" no se encuentra en la base de datos.'
|
41 |
-
contexto = columna.iloc[0]['contexto']
|
42 |
-
|
43 |
-
resultado = {'answer': f'{contexto}'}
|
44 |
-
return resultado['answer']
|
45 |
-
|
46 |
-
def interfaz(nombre, pregunta):
|
47 |
-
origen = obtener_origen(nombre)
|
48 |
-
respuesta = responder_pregunta(nombre, pregunta)
|
49 |
-
return f"Respuesta: {respuesta}\nOrigen: {origen}"
|
50 |
-
|
51 |
-
demo = gr.Interface(
|
52 |
-
fn=interfaz,
|
53 |
-
inputs=[gr.Textbox(label="Nombre del personaje"), gr.Textbox(label="Pregunta")],
|
54 |
-
outputs=gr.Textbox(label="Respuesta"),
|
55 |
-
title="Sistema de Respuesta a Preguntas sobre Personajes Mitol贸gicos",
|
56 |
-
description="Introduce el nombre del personaje y la pregunta (ejem. 驴C贸mo es?, 驴qu茅 hace?...) para obtener una respuesta basada en los datos disponibles."
|
57 |
-
)
|
58 |
-
demo.launch()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|