Update app.py
Browse files
app.py
CHANGED
|
@@ -50,7 +50,7 @@ model_path = hf_hub_download(repo_id=REPO_ID, filename=MODEL_FILENAME)
|
|
| 50 |
|
| 51 |
# Inicializar arquitectura del modelo
|
| 52 |
input_dim = 100 * 100
|
| 53 |
-
dummy_model = VAE(input_dim=input_dim, z_dim=
|
| 54 |
|
| 55 |
# Permitir deserialización segura
|
| 56 |
torch.serialization.add_safe_globals({"VAE": VAE})
|
|
|
|
| 50 |
|
| 51 |
# Inicializar arquitectura del modelo
|
| 52 |
input_dim = 100 * 100
|
| 53 |
+
dummy_model = VAE(input_dim=input_dim, z_dim=40) # la arquitectura base es necesaria para cargar pesos
|
| 54 |
|
| 55 |
# Permitir deserialización segura
|
| 56 |
torch.serialization.add_safe_globals({"VAE": VAE})
|