Update app.py

app.py

@@ -41,9 +41,9 @@ def image_create(seed: int, truncation_psi: float):
             #gaussian=gaussian_radius,
             log_visual_freq=1000
     ))
-    #device = 'cuda' if torch.cuda.is_available() else 'cpu'
-    #model.to(device)
-    #data = data.to(device)
+    device = 'cuda' if torch.cuda.is_available() else 'cpu'
+    model.to(device)
+    data = data.to(device)
     device = th.device()
     generator = create_generator("stylegan2-ffhq-config-f.pt","feng2022/Time-TravelRephotography_stylegan2-ffhq-config-f",args,device)
     #generator = create_generator("checkpoint_b.pt.pth","feng2022/Time_TravelRephotography_checkpoint_b",args,device)