Update model

app.py

@@ -4,14 +4,7 @@ from collections import defaultdict, Counter
 import matplotlib.pyplot as plt
 import gradio as gr
 import pandas as pd
-from transformers import pipeline, AutoTokenizer, AutoModelForTokenClassification
-
-MODELS = ["d4data/biomedical-ner-all"]
-
-current_model = MODELS[0]
-
-tokenizer = AutoTokenizer.from_pretrained(current_model)
-model = AutoModelForTokenClassification.from_pretrained(current_model)
+from transformers import pipeline
 
 plt.switch_backend("Agg")
 
@@ -20,7 +13,11 @@ with open("examples.json", "r") as f:
     content = json.load(f)
     examples = [f"{x['label']}: {x['text']}" for x in content]
 
-pipe = pipeline("ner", model=model, tokenizer=tokenizer, aggregation_strategy="simple")
+pipe = pipeline(
+    "ner",
+    model="Clinical-AI-Apollo/Medical-NER",
+    aggregation_strategy="simple",
+)
 
 
 def plot_to_figure(grouped):