CyberPeace-Institute
/

Cybersecurity-Knowledge-Graph

Token Classification

Model card Files Files and versions

cpi-connect commited on Sep 28, 2023

Commit

29d5112

·

1 Parent(s): 7969312

Upload model

Files changed (1) hide show

model.py +1 -1

model.py CHANGED Viewed

@@ -6,7 +6,7 @@ from sentence_transformers import SentenceTransformer
 from transformers import AutoTokenizer
-from cybersecurity_knowledge_graph.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 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