Update app.py

app.py

@@ -21,7 +21,7 @@ class Generator(nn.Module):
         return self.model(x)
 device = "cpu"
 generator = Generator()
-generator.load_state_dict(torch.load('generator_model.pt'))
+generator.load_state_dict(torch.load('generator_model.pt', map_location=torch.device('cpu')))
 num_faces_to_generate = 10
 z_dim = 13
 ethnicity_map = {'White': 0, 'Black': 1, 'Asian': 2, 'Indian': 3, 'Other': 4}