| import streamlit as st | |
| from transformers import pipeline | |
| 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 | |
| pipe = pipeline('sentiment-analysis') | |
| text = st.text_area('test') | |
| if text: | |
| encoded_input = tokenizer(text, return_tensors='pt') | |
| output = model(**encoded_input) | |
| results = output.logits.detach().cpu().numpy()[0].argsort()[::-1][:5] | |
| vara= [ config.id2label[ids] for ids in results] | |
| st.write(vara) |