will33am commited on
Commit
98bbf5b
1 Parent(s): be2ced2
Files changed (2) hide show
  1. .ipynb_checkpoints/app-checkpoint.py +1 -1
  2. app.py +1 -1
.ipynb_checkpoints/app-checkpoint.py CHANGED
@@ -46,7 +46,7 @@ def gradio_fn(image_input,radio_steps,radio_class,radio_robust):
46
  'iterations': int(radio_steps),
47
  'targeted': True,
48
  'do_tqdm': True,
49
- #'device': DEVICE
50
  }
51
  # Define the target and the image
52
  target = torch.tensor([int(DICT_CLASSES[radio_class])]).to(DEVICE)
 
46
  'iterations': int(radio_steps),
47
  'targeted': True,
48
  'do_tqdm': True,
49
+ 'device': DEVICE
50
  }
51
  # Define the target and the image
52
  target = torch.tensor([int(DICT_CLASSES[radio_class])]).to(DEVICE)
app.py CHANGED
@@ -46,7 +46,7 @@ def gradio_fn(image_input,radio_steps,radio_class,radio_robust):
46
  'iterations': int(radio_steps),
47
  'targeted': True,
48
  'do_tqdm': True,
49
- #'device': DEVICE
50
  }
51
  # Define the target and the image
52
  target = torch.tensor([int(DICT_CLASSES[radio_class])]).to(DEVICE)
 
46
  'iterations': int(radio_steps),
47
  'targeted': True,
48
  'do_tqdm': True,
49
+ 'device': DEVICE
50
  }
51
  # Define the target and the image
52
  target = torch.tensor([int(DICT_CLASSES[radio_class])]).to(DEVICE)