Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 = "
|
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
|