File size: 596 Bytes
ad75dbd 3be88b5 ad75dbd 3be88b5 a460ebb 3be88b5 a460ebb 3be88b5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
import streamlit as st
import utils
from transformers import pipeline
from transformers import AutoTokenizer
from transformers import AutoModelForSequenceClassification
#####################
model_id='hackathon-somos-nlp-2023/DiagTrast'
tokenizer = AutoTokenizer.from_pretrained(model_id)
classifier = pipeline("text-classification",
model=model_id)
#####################
st.title('Diagnóstico de Trastornos Mentales')
sintomas = st.text_input(label = 'Introduce síntomas',
value = '')
st.markdown(classifier(utils.clean_text(sintomas))) |