Update app.py

app.py

@@ -89,7 +89,7 @@ ENTITY_TO_COLOR = {
 
 @st.cache_data
 def get_pipe():
-    model_name = "nassga/nassBioMed"
+    model_name = "nassga/Medical-NER-model-2"
     model = transformers.AutoModelForTokenClassification.from_pretrained(model_name)
     tokenizer = transformers.AutoTokenizer.from_pretrained(model_name)
     pipe = transformers.pipeline("token-classification", model=model, tokenizer=tokenizer, aggregation_strategy="simple")