Hugging Face's logo

Spaces:
ttoosi
/
Generative_Inference_Faces
Sleeping

App Files Community
ttoosi commited on
Commit
15e838d
·
verified ·
1 Parent(s): 15d337f

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -26,7 +26,7 @@ print(f"Using device: {device}")
26
  MODEL_URLS = {
27
  'resnet50_robust': 'https://huggingface.co/madrylab/robust-imagenet-models/resolve/main/resnet50_l2_eps3.ckpt',
28
  'resnet50_standard': 'https://huggingface.co/madrylab/robust-imagenet-models/resolve/main/resnet50_l2_eps0.ckpt',
29
- 'resnet50_robust_face': 'https://huggingface.co/ttoosi/resnet50_robust_face/resolve/main/100_checkpoint.pt'
30
  }
31
 
32
  IMAGENET_MEAN = [0.485, 0.456, 0.406]
 
26
  MODEL_URLS = {
27
  'resnet50_robust': 'https://huggingface.co/madrylab/robust-imagenet-models/resolve/main/resnet50_l2_eps3.ckpt',
28
  'resnet50_standard': 'https://huggingface.co/madrylab/robust-imagenet-models/resolve/main/resnet50_l2_eps0.ckpt',
29
+ 'resnet50_robust_face': 'https://huggingface.co/ttoosi/resnet50_robust_face/resolve/main/resnet50_imagenet_L2_eps_0.50_checkpoint150.pt'
30
  }
31
 
32
  IMAGENET_MEAN = [0.485, 0.456, 0.406]