Update app.py

app.py

@@ -55,7 +55,7 @@ def generate_map(seed: int = None):
     if seed is None:
         seed = torch.randint(10000, (1,)).item()
     torch.manual_seed(seed)
-    z = torch.randn(1, model.latent_dim).to(device)
+    z = torch.randn(1, model.latent_dim).to("cpu")
     with torch.no_grad():
         output = model.decode(z)
     output = sample_with_temperature(output, temperature=3)[0].cpu().numpy()