Update app.py

app.py

@@ -25,8 +25,8 @@ pred = transformers.pipeline("text-classification", model=model,
 
 explainer = shap.Explainer(pred)
 
-ner_tokenizer = AutoTokenizer.from_pretrained("d4data/biomedical-ner-all")
-ner_model = AutoModelForTokenClassification.from_pretrained("d4data/biomedical-ner-all")
+tokenizer = AutoTokenizer.from_pretrained("Clinical-AI-Apollo/Medical-NER")
+model = AutoModelForTokenClassification.from_pretrained("Clinical-AI-Apollo/Medical-NER")
 ner_pipe = pipeline("ner", model=ner_model, tokenizer=ner_tokenizer, aggregation_strategy="simple")
 
 def adr_predict(x):