Update inference.py

testing the chekcpoint I trained

inference.py

@@ -24,7 +24,8 @@ print(f"Using device: {device}")
 
 # Constants
 MODEL_URLS = {
-    'resnet50_robust': 'https://huggingface.co/madrylab/robust-imagenet-models/resolve/main/resnet50_l2_eps3.ckpt',
+    # 'resnet50_robust': 'https://huggingface.co/madrylab/robust-imagenet-models/resolve/main/resnet50_l2_eps3.ckpt',
+    'resnet50_robust':'https://huggingface.co/ttoosi/robust_resnet50_eps_3.00_epoch_best/blob/main/checkpoint_best.pt',
     'resnet50_standard': 'https://huggingface.co/madrylab/robust-imagenet-models/resolve/main/resnet50_l2_eps0.ckpt',
     'resnet50_robust_face': 'https://huggingface.co/ttoosi/resnet50_robust_face/blob/main/100_checkpoint.pt'
 }