import streamlit as st
from transformers import AutoTokenizer, BertForSequenceClassification

tokenizer = AutoTokenizer.from_pretrained("AkshatSurolia/ICD-10-Code-Prediction")

model = BertForSequenceClassification.from_pretrained("AkshatSurolia/ICD-10-Code-Prediction")

config = model.config

text = st.text_input("diagnóstico", "Diabetes mellitus")

encoded_input = tokenizer(text, return_tensors='pt')
output = model(**encoded_input)

st.write(output)