jcmachicao commited on
Commit
a3d534a
1 Parent(s): b2bdd91
Files changed (1) hide show
  1. app.py +31 -0
app.py ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #python file
2
+ #jcmachicao
3
+
4
+ import streamlit as st
5
+ from transformers import pipeline
6
+ from transformers import AutoTokenizer, AutoModelForQuestionAnswering
7
+ tokenizer = AutoTokenizer.from_pretrained("mrm8488/bert-base-spanish-wwm-cased-finetuned-spa-squad2-es")
8
+ modelo_qa = AutoModelForQuestionAnswering.from_pretrained("mrm8488/bert-base-spanish-wwm-cased-finetuned-spa-squad2-es")
9
+ preguntas = pipeline('question-answering', model=modelo_qa, tokenizer=tokenizer)
10
+
11
+ x = st.slider('Selecciona un valor: ')
12
+ st.write('El cuadrado de ', x, 'es ', x * x)
13
+
14
+ texto_default = 'Los jardines tienen flores que dependen del cuidado de los jardineros. Una buena selección de los jardineros es garantía de la calidad de las flores.'
15
+ texto = st.text_area('Cargue un texto de hasta 5 hojas, 1500 palabras, ó 15000 caracteres: ', texto_default)
16
+
17
+ preg_1 = st.text_input('Pregunta 1', '¿Qué son los jardines?')
18
+ preg_2 = st.text_input('Pregunta 2', '¿Qué garantiza la calidad de las flores?')
19
+ preg_3 = st.text_input('Pregunta 3', '¿Qué relación hay entre las flores y los jardineros?')
20
+
21
+ pregs = [preg_1, preg_2, preg_3]
22
+
23
+ boton_preg2 = st.button('Enviar preguntas')
24
+
25
+ if boton_preg2:
26
+ for preg in pregs:
27
+ respuesta = preguntas({'question': preg, 'context': texto})
28
+ st.write(preg)
29
+ st.write(respuesta['answer'])
30
+ st.write('\n Confiabilidad: ', respuesta['score'])
31
+