Update app.py

app.py

@@ -43,7 +43,7 @@ class VAE(nn.Module):
 # 2. CARGAR EL MODELO DESDE HUGGING FACE
 # -----------------------------
 REPO_ID = "Bmo411/VAE"  # <-- reemplaza con tu repo si cambia
-MODEL_FILENAME = "vae_complete_model.pth"
+MODEL_FILENAME = "vae_complete_model (1).pth"
 
 # Descargar modelo automáticamente
 model_path = hf_hub_download(repo_id=REPO_ID, filename=MODEL_FILENAME)
@@ -69,7 +69,7 @@ def generate_image():
     with torch.no_grad():
         z = torch.randn(1, z_dim).to(device)
         out = model.decode(z)
-        out = out.view(1, 1, 28, 28)
+        out = out.view(1, 1, 100, 100)
 
         output_path = "generated_sample.png"
         save_image(out, output_path)