marcelomoreno26 commited on
Commit
c7cbc23
1 Parent(s): cd28c0f

Update model_functions.py

Browse files
Files changed (1) hide show
  1. model_functions.py +3 -3
model_functions.py CHANGED
@@ -24,9 +24,9 @@ def load_summarizer():
24
  return tokenizer, model
25
 
26
  def load_NER():
27
- config = AutoConfig.from_pretrained("hannahisrael03/distilbert-base-uncased-finetuned-wikiann")
28
- model = AutoModelForTokenClassification.from_pretrained("hannahisrael03/distilbert-base-uncased-finetuned-wikiann",config=config)
29
- tokenizer = AutoTokenizer.from_pretrained("hannahisrael03/distilbert-base-uncased-finetuned-wikiann")
30
  pipe = pipeline("ner", model=model, tokenizer=tokenizer, aggregation_strategy="average")
31
 
32
  return pipe
 
24
  return tokenizer, model
25
 
26
  def load_NER():
27
+ config = AutoConfig.from_pretrained("hannahisrael03/wikineural-multilingual-ner-finetuned-wikiann")
28
+ model = AutoModelForTokenClassification.from_pretrained("hannahisrael03/wikineural-multilingual-ner-finetuned-wikiann",config=config)
29
+ tokenizer = AutoTokenizer.from_pretrained("hannahisrael03/wikineural-multilingual-ner-finetuned-wikiann")
30
  pipe = pipeline("ner", model=model, tokenizer=tokenizer, aggregation_strategy="average")
31
 
32
  return pipe