Jimenagh commited on
Commit
e57a01a
verified
1 Parent(s): 29366bf

Delete app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -58
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()