|
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 = 'Nadie me quiere') |
|
|
|
st.markdown(classifier(utils.clean_text(sintomas[0]['label']))) |
