Update README.md

README.md

@@ -117,13 +117,7 @@ NUM_LABELS = { 'age': len(age_labels), 'gender': len(gender_labels), 'race': len
 processor = CLIPImageProcessor.from_pretrained(MODEL_PATH)
 model = MultiTaskClipVisionModel(num_labels=NUM_LABELS)
 
-# Load the fine-tuned weights from the hub
-model.load_state_dict(
-    torch.hub.load_state_dict_from_url(
-        f"https://huggingface.co/{MODEL_PATH}/resolve/main/pytorch_model.bin",
-        map_location=DEVICE
-    )
-)
+
 model.to(DEVICE)
 model.eval()