Update app.py

app.py

@@ -119,7 +119,7 @@ def main():
   tokenizer = DistilBertTokenizerFast.from_pretrained(model_name)
   model_path1 = "DistilBertforTokenclassification"
   
-  model = DistilBertForTokenClassification.from_pretrained(model_path1, id2label={0:'CT',1:'E',2:'C',3:'O'}) #len(unique_tags),, num_labels= 7,
+  model = DistilBertForTokenClassification.from_pretrained(model_path1) #len(unique_tags),, num_labels= 7, , id2label={0:'CT',1:'E',2:'C',3:'O'}
   pipe = pipeline('ner', model=model, tokenizer=tokenizer,aggregation_strategy='simple') #grouped_entities=True
 
   sentence_pred = []