BabakScrapes commited on
Commit
9124b26
·
verified ·
1 Parent(s): 132623b

Update pipeline.py

Browse files
Files changed (1) hide show
  1. 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("{}\\Trained Models\\clause_model_512".format(dir_path), num_labels=3)
21
- classification_model = RobertaForSequenceClassification.from_pretrained("{}\Trained Models\classfication_model".format(dir_path), num_labels=18)
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"),