Spaces:
Runtime error
Runtime error
File size: 456 Bytes
473643d 33d2692 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
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) |