to_cpu
Browse files- __pycache__/helpers.cpython-310.pyc +0 -0
- app.py +2 -2
__pycache__/helpers.cpython-310.pyc
ADDED
|
Binary file (8.19 kB). View file
|
|
|
app.py
CHANGED
|
@@ -649,8 +649,8 @@ from torch.utils.data import DataLoader
|
|
| 649 |
from helpers import get_imagenet_dataset, get_torchvision_model, denormalize_image
|
| 650 |
|
| 651 |
os.environ['GRADIO_TEMP_DIR'] = '.tmp'
|
| 652 |
-
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 653 |
-
|
| 654 |
|
| 655 |
def target_visualizer(x): return dataset.dataset.idx_to_label(x.item())
|
| 656 |
|
|
|
|
| 649 |
from helpers import get_imagenet_dataset, get_torchvision_model, denormalize_image
|
| 650 |
|
| 651 |
os.environ['GRADIO_TEMP_DIR'] = '.tmp'
|
| 652 |
+
# device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 653 |
+
device = torch.device("cpu")
|
| 654 |
|
| 655 |
def target_visualizer(x): return dataset.dataset.idx_to_label(x.item())
|
| 656 |
|