| 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))) |