ejqs commited on
Commit
8973d97
·
verified ·
1 Parent(s): f5b5999

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -8,7 +8,7 @@ class EndpointHandler():
8
 
9
  # Load the models -- Leadership
10
  self.leadership_model_name = "roberta-large-mnli"
11
- self.leadership_model_path = path + "/pet-leadership-model-roberta-large-mnli_bs4_gas4_lr1e-05_ep5" + "/checkpoint-1560"
12
  self.leadership_pattern = "Sentence: {} Question: Does this show leadership? Answer: <mask>"
13
 
14
  # Load the models -- Collaboration
 
8
 
9
  # Load the models -- Leadership
10
  self.leadership_model_name = "roberta-large-mnli"
11
+ self.leadership_model_path = path + "/pet-leadership-model-roberta-large-mnli_bs4_gas4_lr1e-05_ep5" + "/checkpoint-1855"
12
  self.leadership_pattern = "Sentence: {} Question: Does this show leadership? Answer: <mask>"
13
 
14
  # Load the models -- Collaboration