Update app.py

app.py

@@ -22,9 +22,9 @@ def generate_images (initial_archi, initial_class, batch_size):
     class_vector = torch.from_numpy(class_vector)
 
     # If you have a GPU, put everything on cuda
-    noise_vector = noise_vector.to('cuda')
-    class_vector = class_vector.to('cuda')
-    gan_model.to('cuda')
+    #noise_vector = noise_vector.to('cuda')
+    #class_vector = class_vector.to('cuda')
+    #gan_model.to('cuda')
 
     # Generate an image
     with torch.no_grad():