Bmo411 commited on
Commit
249c35f
·
verified ·
1 Parent(s): ada5b25

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=20) # la arquitectura base es necesaria para cargar pesos
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})