Upload model

model.py

@@ -7,14 +7,14 @@ from transformers import AutoTokenizer
 
 
 from .nugget_model_utils import CustomRobertaWithPOS as NuggetModel
-from cybersecurity_knowledge_graph.args_model_utils import CustomRobertaWithPOS as ArgumentModel
-from cybersecurity_knowledge_graph.realis_model_utils import CustomRobertaWithPOS as RealisModel
+from .args_model_utils import CustomRobertaWithPOS as ArgumentModel
+from .realis_model_utils import CustomRobertaWithPOS as RealisModel
 
 from .configuration import CybersecurityKnowledgeGraphConfig
 
-from cybersecurity_knowledge_graph.event_nugget_predict import create_dataloader as event_nugget_dataloader
-from cybersecurity_knowledge_graph.event_realis_predict import create_dataloader as event_realis_dataloader
-from cybersecurity_knowledge_graph.event_arg_predict import create_dataloader as event_argument_dataloader
+from .event_nugget_predict import create_dataloader as event_nugget_dataloader
+from .event_realis_predict import create_dataloader as event_realis_dataloader
+from .event_arg_predict import create_dataloader as event_argument_dataloader
 
 class CybersecurityKnowledgeGraphModel(PreTrainedModel):
     config_class = CybersecurityKnowledgeGraphConfig