Update app.py

app.py

@@ -23,7 +23,7 @@ class Generator(nn.Module):
 latent_dim = 100
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 generator = Generator(latent_dim).to(device)
-generator_model_path = 'noisyKickGAN/generator_model.pkl'
+generator_model_path = 'generator_model.pkl'
 generator.load_state_dict(torch.load(generator_model_path, map_location=device))