Update app.py

app.py

@@ -110,7 +110,7 @@ def main():
     
   tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased") #bert-base-uncased
   
-  model_path = "checkpoint2850"
+  model_path = "checkpoint-2850"
   
   model = AutoModelForSequenceClassification.from_pretrained(model_path,id2label={0:'non-causal',1:'causal'})
 
@@ -126,7 +126,7 @@ def main():
   
   
   
-  model_path1 = "DistilBertForTokeClassification"
+  model_path1 = "DistilBertforTokenclassification"
   
   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