Update handler.py

handler.py

@@ -8,13 +8,13 @@ class EndpointHandler():
 
         # Load the models -- Leadership
         self.leadership_model_name = "roberta-large-mnli"
-        self.leadership_model_path = path + "/pet-leadership-model-roberta-large-mnli_bs4_gas4_lr3e-05_ep2" + "/checkpoint-742"
+        self.leadership_model_path = path + "/pet-leadership-model-roberta-large-mnli_bs4_gas4_lr1e-05_ep5" + "/checkpoint-1560"
         self.leadership_pattern = "Sentence: {} Question: Does this show leadership? Answer: <mask>"
 
         # Load the models -- Collaboration
-        self.collab_model_name = "facebook/bart-large-mnli"
-        self.collab_model_path = path + "/pet-collaboration-model-facebook-bart-large-mnli_bs4_gas4_lr5e-05_ep3" + "/checkpoint-936"
-        self.collab_pattern = "Sentence: {} Question: Does this show teamwork and collaboration? Answer: <mask>"
+        self.collab_model_name = "roberta-large-mnli"
+        self.collab_model_path = path + "/pet-collaboration-model-roberta-large-mnli_bs4_gas4_lr1e-05_ep5" + "/checkpoint-1560"
+        self.collab_pattern = "Sentence: {} Question: Does this show teamwork? Answer: <mask>"
 
         # Load the tokenizer
         self.leadership_tokenizer = AutoTokenizer.from_pretrained(self.leadership_model_name)