Update pipeline.py
Browse files- pipeline.py +2 -2
pipeline.py
CHANGED
@@ -17,8 +17,8 @@ os.environ["KMP_DUPLICATE_LIB_OK"]="TRUE"
|
|
17 |
os.system("python -m spacy download en_core_web_sm")
|
18 |
nlp = spacy.load("en_core_web_sm")
|
19 |
tokenizer = RobertaTokenizer.from_pretrained("roberta-base")
|
20 |
-
clause_model = AutoModelForTokenClassification.from_pretrained("{}
|
21 |
-
classification_model = RobertaForSequenceClassification.from_pretrained("{}
|
22 |
|
23 |
labels2attrs = {
|
24 |
"##BOUNDED EVENT (SPECIFIC)": ("specific", "dynamic", "episodic"),
|
|
|
17 |
os.system("python -m spacy download en_core_web_sm")
|
18 |
nlp = spacy.load("en_core_web_sm")
|
19 |
tokenizer = RobertaTokenizer.from_pretrained("roberta-base")
|
20 |
+
clause_model = AutoModelForTokenClassification.from_pretrained("{}/clause_model_512".format(dir_path), num_labels=3)
|
21 |
+
classification_model = RobertaForSequenceClassification.from_pretrained("{}/classfication_model".format(dir_path), num_labels=18)
|
22 |
|
23 |
labels2attrs = {
|
24 |
"##BOUNDED EVENT (SPECIFIC)": ("specific", "dynamic", "episodic"),
|