Update app.py

app.py

@@ -29,7 +29,7 @@ gender_map = {'Male': 0, 'Female': 1}
 
 def generate_faces(age, ethnicity, gender):
     
-    random_z = torch.randn(num_faces_to_generate, z_dim).to(device)
+    random_z = torch.randn(num_faces_to_generate, z_dim)
 
     
     random_z[:, 0] = age
@@ -37,7 +37,7 @@ def generate_faces(age, ethnicity, gender):
     random_z[:, 2] = gender_map[gender] 
 
     
-    random_z[:, 3:] = torch.randn(num_faces_to_generate, z_dim - 3).to(device)
+    random_z[:, 3:] = torch.randn(num_faces_to_generate, z_dim - 3)
     with torch.no_grad():
         generated_faces = generator(random_z)