Hugging Face's logo

Spaces:
ttoosi
/
GenerativeInferenceDemo
Running on Zero

App Files Community
1
ttoosi commited on
Commit
4752d1b
·
verified ·
1 Parent(s): ff6ea5e

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +2 -3
inference.py CHANGED
@@ -18,10 +18,9 @@ print(f"Using device: {device}")
18
 
19
  # Constants
20
  MODEL_URLS = {
21
- 'robust_resnet50': 'https://huggingface.co/madrylab/robust-imagenet-models/resolve/main/resnet50_l2_eps_3.0.pt',
22
- 'standard_resnet50': 'https://huggingface.co/madrylab/robust-imagenet-models/resolve/main/resnet50_l2_eps_0.0.pt'
23
  }
24
-
25
  IMAGENET_MEAN = [0.485, 0.456, 0.406]
26
  IMAGENET_STD = [0.229, 0.224, 0.225]
27
 
 
18
 
19
  # Constants
20
  MODEL_URLS = {
21
+ 'robust_resnet50': 'https://huggingface.co/madrylab/robust-imagenet-models/resolve/main/resnet50_l2_eps3.ckpt',
22
+ 'standard_resnet50': 'https://huggingface.co/madrylab/robust-imagenet-models/resolve/main/resnet50_l2_eps0.ckpt'
23
  }
 
24
  IMAGENET_MEAN = [0.485, 0.456, 0.406]
25
  IMAGENET_STD = [0.229, 0.224, 0.225]
26