nassga commited on
Commit
eeb16df
·
1 Parent(s): e86afe8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -89,7 +89,7 @@ ENTITY_TO_COLOR = {
89
 
90
  @st.cache_data
91
  def get_pipe():
92
- model_name = "nassga/NassBioMedical"
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")