Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -89,7 +89,7 @@ ENTITY_TO_COLOR = {
|
|
89 |
|
90 |
@st.cache_data
|
91 |
def get_pipe():
|
92 |
-
model_name = "nassga/
|
93 |
model = transformers.AutoModelForTokenClassification.from_pretrained(model_name)
|
94 |
tokenizer = transformers.AutoTokenizer.from_pretrained(model_name)
|
95 |
pipe = transformers.pipeline("token-classification", model=model, tokenizer=tokenizer, aggregation_strategy="simple")
|
|
|
89 |
|
90 |
@st.cache_data
|
91 |
def get_pipe():
|
92 |
+
model_name = "nassga/nassGanBioMedical"
|
93 |
model = transformers.AutoModelForTokenClassification.from_pretrained(model_name)
|
94 |
tokenizer = transformers.AutoTokenizer.from_pretrained(model_name)
|
95 |
pipe = transformers.pipeline("token-classification", model=model, tokenizer=tokenizer, aggregation_strategy="simple")
|