Spaces:

Souranil
/

VAE

Build error

Souranil commited on Apr 12, 2022

Commit

8d21e52

1 Parent(s): a1e64a3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ st.markdown(
 )
 def load_model_files():
-    files = os.listdir("./saved_models/")
     # Docker creates some whiteout files which mig
     files = [i for i in files if ".ckpt" in i]
     clean_names = [utils.parse_model_file_name(name) for name in files]

 )
 def load_model_files():
+    files = os.listdir("./models/")
     # Docker creates some whiteout files which mig
     files = [i for i in files if ".ckpt" in i]
     clean_names = [utils.parse_model_file_name(name) for name in files]