Update README.md

README.md

@@ -58,6 +58,8 @@ import torch.nn as nn
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch.optim as optim
 import numpy as np
+from huggingface_hub import hf_hub_download
+import json
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 # Load explicitly your fine-tuned MPNet model
@@ -66,8 +68,6 @@ classifier_model = AutoModelForSequenceClassification.from_pretrained("selfconst
 # Load explicitly your tokenizer
 tokenizer = AutoTokenizer.from_pretrained("selfconstruct3d/AttackGroup-MPNET")
 
-from huggingface_hub import hf_hub_download
-import json
 
 label_to_groupid_file = hf_hub_download(
     repo_id="selfconstruct3d/AttackGroup-MPNET",