Upload model

model.py

@@ -35,21 +35,7 @@ class CybersecurityKnowledgeGraphModel(PreTrainedModel):
         self.event_argument_model = ArgumentModel(num_classes = 43)
         self.event_realis_model = RealisModel(num_classes_realis = 4)
 
-        self.event_nugget_model.load_state_dict(torch.load(self.event_nugget_model_path))
-        self.event_realis_model.load_state_dict(torch.load(self.event_realis_model_path))
-        self.event_argument_model.load_state_dict(torch.load(self.event_argument_model_path))
-
-        role_classifiers = {}
-        folder_path = '/cybersecurity_knowledge_graph/arg_role_models'
-
-        for filename in os.listdir(os.getcwd() + folder_path):
-            if filename.endswith('.joblib'):
-                file_path = os.getcwd() + os.path.join(folder_path, filename)
-                clf = joblib.load(file_path)
-                arg = filename.split(".")[0]
-                role_classifiers[arg] = clf
-
-        self.role_classifiers = role_classifiers
+        self.role_classifiers = None
         self.embed_model = SentenceTransformer('all-MiniLM-L6-v2')