Hugging Face's logo

Spaces:
Warvito
/
diffusion_brain
Build error

App Files Community
Warvito commited on
Commit
387251c
·
1 Parent(s): 6de39a8

Fix paths

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -20,12 +20,12 @@ mediapy.set_ffmpeg(ffmpeg_path)
20
 
21
  # Loading model
22
  vqvae = mlflow.pytorch.load_model(
23
- "./trained_models/vae/final_model"
24
  )
25
  vqvae.eval()
26
 
27
  diffusion = mlflow.pytorch.load_model(
28
- "./trained_models/ddpm/final_model"
29
  )
30
  diffusion.eval()
31
 
 
20
 
21
  # Loading model
22
  vqvae = mlflow.pytorch.load_model(
23
+ "./trained_models/vae/"
24
  )
25
  vqvae.eval()
26
 
27
  diffusion = mlflow.pytorch.load_model(
28
+ "./trained_models/ddpm/"
29
  )
30
  diffusion.eval()
31