Update app.py

app.py

@@ -1,9 +1,11 @@
+import strimlit 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 = "subarachnoid hemorrhage scalp laceration service: surgery major surgical or invasive"
+text = st.text_area('enter some text!')
+#text = "subarachnoid hemorrhage scalp laceration service: surgery major surgical or invasive"
 encoded_input = tokenizer(text, return_tensors='pt')
 output = model(**encoded_input)
 
@@ -11,8 +13,8 @@ results = output.logits.detach().cpu().numpy()[0].argsort()[::-1][:5]
 #return [ config.id2label[ids] for ids in results]
 
 #pipe = pipeline('sentiment-analysis')
-#text = st.text_area('enter some text!')
 
-#if text:
-#  out = pipe(text)
-#  st.json(out)
+
+if text:
+  out = pipe(text)
+  st.json('{}')