Hugging Face's logo

Spaces:
ttoosi
/
GenerativeInferenceDemo
Running on Zero

App Files Community
1
ttoosi commited on
Commit
2b5cd8c
·
verified ·
1 Parent(s): 11a7475

to resnet50_robust

Browse files
Files changed (1) hide show
  1. inference.py +2 -2
inference.py CHANGED
@@ -24,8 +24,8 @@ print(f"Using device: {device}")
24
 
25
  # Constants
26
  MODEL_URLS = {
27
- 'robust_resnet50': 'https://huggingface.co/madrylab/robust-imagenet-models/resolve/main/resnet50_l2_eps3.ckpt',
28
- 'standard_resnet50': 'https://huggingface.co/madrylab/robust-imagenet-models/resolve/main/resnet50_l2_eps0.ckpt'
29
  }
30
 
31
  IMAGENET_MEAN = [0.485, 0.456, 0.406]
 
24
 
25
  # Constants
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
  }
30
 
31
  IMAGENET_MEAN = [0.485, 0.456, 0.406]