Update app.py

app.py

@@ -10,7 +10,7 @@ import torch.nn as nn
 # 1. DEFINICIÓN DEL MODELO VAE
 # -----------------------------
 class VAE(nn.Module):
-    def __init__(self, input_dim, h_dim=400, z_dim=20):  # NOTA: z_dim por defecto en 20
+    def __init__(self, input_dim, h_dim=400, z_dim=40):  # NOTA: z_dim por defecto en 20
         super().__init__()
         # Encoder
         self.img_2hid = nn.Linear(input_dim, h_dim)