Update app.py
Browse files
app.py
CHANGED
|
@@ -60,6 +60,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
|
| 60 |
model = torch.load(model_path, map_location=device, weights_only=False)
|
| 61 |
model.to(device)
|
| 62 |
model.eval()
|
|
|
|
| 63 |
|
| 64 |
# -----------------------------
|
| 65 |
# 3. FUNCIÓN PARA GENERAR IMAGEN
|
|
|
|
| 60 |
model = torch.load(model_path, map_location=device, weights_only=False)
|
| 61 |
model.to(device)
|
| 62 |
model.eval()
|
| 63 |
+
z_dim = model.z_2hid.in_features
|
| 64 |
|
| 65 |
# -----------------------------
|
| 66 |
# 3. FUNCIÓN PARA GENERAR IMAGEN
|