Update app.py

app.py

@@ -20,8 +20,8 @@ noise_vector = torch.from_numpy(noise_vector)
 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')
+#noise_vector = noise_vector.to('cuda')
+#class_vector = class_vector.to('cuda')
 #gan_model.to('cuda')
 
 # Generate an image