Update handler.py
Browse files- 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-
|
| 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
|