Spaces:
Runtime error
Runtime error
| 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) |