Spaces:

kingav
/

generative-playground

Sleeping

kingav commited on Jun 26, 2024

Commit

28476eb

verified ·

1 Parent(s): dd6885d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -156,7 +156,8 @@ if option == "Auto-Regressor":
         # %%
         import os
-        checkpoint_path = "training_1/cp.ckpt"
         checkpoint_dir = os.path.dirname(checkpoint_path)

         # %%
         import os
+        # checkpoint_path = "training_1/cp.ckpt"
+        checkpoint_path = "training_1/cp.weights.h5"
         checkpoint_dir = os.path.dirname(checkpoint_path)