Create app.py

app.py

@@ -0,0 +1,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(text)