Spaces:
Running
Running
andrewgleave
commited on
Commit
·
a9898b9
1
Parent(s):
621a7b7
Remove old model
Browse files
app.py
CHANGED
@@ -6,9 +6,9 @@ import gradio as gr
|
|
6 |
import pandas as pd
|
7 |
from transformers import pipeline, AutoTokenizer, AutoModelForTokenClassification
|
8 |
|
9 |
-
MODELS = ["d4data/biomedical-ner-all"
|
10 |
|
11 |
-
current_model = MODELS[
|
12 |
|
13 |
tokenizer = AutoTokenizer.from_pretrained(current_model)
|
14 |
model = AutoModelForTokenClassification.from_pretrained(current_model)
|
|
|
6 |
import pandas as pd
|
7 |
from transformers import pipeline, AutoTokenizer, AutoModelForTokenClassification
|
8 |
|
9 |
+
MODELS = ["d4data/biomedical-ner-all"]
|
10 |
|
11 |
+
current_model = MODELS[0]
|
12 |
|
13 |
tokenizer = AutoTokenizer.from_pretrained(current_model)
|
14 |
model = AutoModelForTokenClassification.from_pretrained(current_model)
|