Update app.py

app.py

@@ -58,7 +58,7 @@ def genreate_intepriable_output(input,dataset="CUB2011", arch="resnet50",seed=12
             transforms.ToTensor(),
         ])
     TR=get_augmentation(0.1, img_size, False, False, True, True, normalize_params["CUB2011"])
-    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+    device = torch.device("cpu")
     if folder is None:
         folder = Path(f"tmp/{arch}/{dataset}/{seed}/")