inesani commited on
Commit
3584ed4
1 Parent(s): 4a4563a

back to original ner-log model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ auth_token = os.environ["TOKEN"]
7
 
8
  tokenizer = DistilBertTokenizerFast.from_pretrained('distilbert-base-uncased')
9
 
10
- model_path = "inesani/ner-log-extended"
11
  model_loaded = DistilBertForTokenClassification.from_pretrained(model_path, use_auth_token=auth_token)
12
 
13
  ner_pipeline = pipeline("ner", model=model_loaded, tokenizer=tokenizer,
 
7
 
8
  tokenizer = DistilBertTokenizerFast.from_pretrained('distilbert-base-uncased')
9
 
10
+ model_path = "inesani/ner-log"
11
  model_loaded = DistilBertForTokenClassification.from_pretrained(model_path, use_auth_token=auth_token)
12
 
13
  ner_pipeline = pipeline("ner", model=model_loaded, tokenizer=tokenizer,