Spaces:
Build error
Build error
update
Browse files- .ipynb_checkpoints/app-checkpoint.py +1 -1
- 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 |
-
|
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 |
-
|
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)
|