Seetha commited on
Commit
d6fe52a
1 Parent(s): cf8f017

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -117,7 +117,7 @@ def main():
117
 
118
  model_name = "distilbert-base-uncased"
119
  tokenizer = DistilBertTokenizerFast.from_pretrained(model_name)
120
- model_path1 = "DistilBertforTokenClassification"
121
 
122
  model = DistilBertForTokenClassification.from_pretrained(model_path1, id2label={0:'CT',1:'E',2:'C',3:'O'}) #len(unique_tags),, num_labels= 7,
123
  pipe = pipeline('ner', model=model, tokenizer=tokenizer,aggregation_strategy='simple') #grouped_entities=True
 
117
 
118
  model_name = "distilbert-base-uncased"
119
  tokenizer = DistilBertTokenizerFast.from_pretrained(model_name)
120
+ model_path1 = "DistilBertforTokenclassification"
121
 
122
  model = DistilBertForTokenClassification.from_pretrained(model_path1, id2label={0:'CT',1:'E',2:'C',3:'O'}) #len(unique_tags),, num_labels= 7,
123
  pipe = pipeline('ner', model=model, tokenizer=tokenizer,aggregation_strategy='simple') #grouped_entities=True