Spaces:

ttoosi
/

Generative_Inference_Faces

Sleeping

ttoosi commited on Nov 9

Commit

226318e

verified ·

1 Parent(s): f9d3bad

Update inference.py

Files changed (1) hide show

inference.py CHANGED Viewed

@@ -29,7 +29,7 @@ print(f"Using device: {device}")
 MODEL_URLS = {
     'resnet50_robust': 'https://huggingface.co/madrylab/robust-imagenet-models/resolve/main/resnet50_l2_eps3.ckpt',
     '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/resolve/main/resnet50_imagenet_L2_eps_0.50_checkpoint150.pt'
 }
 # Model-specific preprocessing configurations

 MODEL_URLS = {
     'resnet50_robust': 'https://huggingface.co/madrylab/robust-imagenet-models/resolve/main/resnet50_l2_eps3.ckpt',
     '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/resolve/main/resnet50_vggface2_L2_eps_0.50_checkpoint150.pt'
 }
 # Model-specific preprocessing configurations