Spaces:
Runtime error
Runtime error
File size: 2,128 Bytes
df486f0 a3d534a 96ede9b a3d534a 0ac3a3f 97ad8e8 9c0c26e 97ad8e8 a3d534a e75324c 6b303c9 a3d534a 5a8ee31 7ea269c a3d534a 12cef0f a8d2364 12cef0f e75324c 12cef0f 02c5f99 12cef0f 88ffa88 12cef0f a8d2364 88ffa88 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 |
# -*- coding: utf-8 -*-
'''
José Carlos Machicao
GestioDinámica
Fecha de producción: 2022_01_31
Fecha de actualización 2022_02_15
'''
import streamlit as st
import torch
from transformers import pipeline
from transformers import AutoTokenizer, AutoModelForQuestionAnswering
st.image('gdmk.png', width=150)
texto = st.text_area('Cargue un texto de hasta 2 hojas, 500 palabras, ó 5000 caracteres: ', height=100)
st.write('Si no tienes un texto listo, puedes usar el que está en este link, sólo copiando y pegando el texto en el espacio de arriba.')
st.write('[link](https://huggingface.co/spaces/jcmachicao/dialogatexto/blob/main/trucha_peru.txt)')
st.write('También puedes modificar las preguntas.')
preg_1 = st.text_input('Pregunta 1', '¿Cuánta trucha exportó el Perú durante el año 2020?')
preg_2 = st.text_input('Pregunta 2', '¿Qué países estuvieron involucrados?')
preg_3 = st.text_input('Pregunta 3', '¿Cómo interviene el proceso de congelamiento?')
pregs = [preg_1, preg_2, preg_3]
if len(pregs)<3:
st.write('Por favor complete las 3 preguntas')
boton_preg = st.button('Enviar preguntas')
modelo_hf2 = 'mrm8488/distill-bert-base-spanish-wwm-cased-finetuned-spa-squad2-es'
modelo_qa = AutoModelForQuestionAnswering.from_pretrained(modelo_hf2)
#modelo_qa = AutoModelForQuestionAnswering.from_pretrained("mrm8488/bert-base-spanish-wwm-cased-finetuned-spa-squad2-es")
tokenizer = AutoTokenizer.from_pretrained("mrm8488/bert-base-spanish-wwm-cased-finetuned-spa-squad2-es",
#{"use_fast": False}
)
tokenizer2 = AutoTokenizer.from_pretrained(modelo_hf2,
#{"use_fast": False}
)
if boton_preg:
st.write('Procesando texto ', texto[0:50], '...')
preguntas = pipeline('question-answering', model=modelo_qa, tokenizer=tokenizer2)
for preg in pregs:
respuesta = preguntas({'question': preg, 'context': texto})
st.write(preg)
st.write(respuesta['answer'])
st.write('\n Confiabilidad: ', respuesta['score'])
|