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
|