Update README.md

README.md

@@ -28,9 +28,8 @@ model_name = "yasserrmd/nsfk-detection"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSequenceClassification.from_pretrained(model_name)
 
-# Load label map
-with open('./results/checkpoint-4467/label_map.json', 'r') as f:
-    label_map = json.load(f)
+label_map = {"True": 0, "False": 1}
+
 id_to_label = {i: label for label, i in label_map.items()}
 
 threshold = 0.7  # Confidence threshold for classification