nouf-sst commited on
Commit
5cd173e
·
1 Parent(s): 8323956
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -308,7 +308,7 @@ def check_softgoal_syntax(softgoals):
308
  # ########## Incorrect Task Syntax ###########
309
  def find_non_VPs(sentences):
310
 
311
- pipeline = TokenClassificationPipeline(model=pos_modelmodel, tokenizer=pos_tokenizertokenizer)
312
 
313
  outputs = pipeline(sentences)
314
 
@@ -402,9 +402,9 @@ def do_nli(premise, hypothesis):
402
  seg_ids = []
403
  mask_ids = []
404
 
405
- premise_id = nli_tokenizertokenizer.encode(premise, add_special_tokens = False)
406
- hypothesis_id = nli_tokenizertokenizer.encode(hypothesis, add_special_tokens = False)
407
- pair_token_ids = [nli_tokenizertokenizer.cls_token_id] + premise_id + [nli_tokenizertokenizer.sep_token_id] + hypothesis_id + [nli_tokenizertokenizer.sep_token_id]
408
  premise_len = len(premise_id)
409
  hypothesis_len = len(hypothesis_id)
410
 
 
308
  # ########## Incorrect Task Syntax ###########
309
  def find_non_VPs(sentences):
310
 
311
+ pipeline = TokenClassificationPipeline(model=pos_model, tokenizer=pos_tokenizer)
312
 
313
  outputs = pipeline(sentences)
314
 
 
402
  seg_ids = []
403
  mask_ids = []
404
 
405
+ premise_id = nli_tokenizer.encode(premise, add_special_tokens = False)
406
+ hypothesis_id = nli_tokenizer.encode(hypothesis, add_special_tokens = False)
407
+ pair_token_ids = [nli_tokenizer.cls_token_id] + premise_id + [nli_tokenizer.sep_token_id] + hypothesis_id + [nli_tokenizer.sep_token_id]
408
  premise_len = len(premise_id)
409
  hypothesis_len = len(hypothesis_id)
410