Spaces:

sayakpaul
/

pokemon-sd-kerascv

Build error

chansung commited on Jan 3, 2023

Commit

edc88c2

•

1 Parent(s): c0cd2c4

update ckpt pointer

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,8 @@ from share_btn import community_icon_html, loading_icon_html, share_js
 # Load model.
 weights_path = keras.utils.get_file(
-    origin="https://storage.googleapis.com/sd-weights-kerascv/woga/ckpt_epochs_72_res_512_mp_True.h5"
 )
 pokemon_model = keras_cv.models.StableDiffusion(
     img_width=img_width, img_height=img_height

 # Load model.
 weights_path = keras.utils.get_file(
+    origin="https://huggingface.co/sayakpaul/kerascv_sd_pokemon_finetuned/raw/main/ckpt_epochs_72_res_512_mp_True.h5",
+    file_hash="10b20bd27912d1da904dafe8c576351c2f373546f446591aeff00d816d701a6e"
 )
 pokemon_model = keras_cv.models.StableDiffusion(
     img_width=img_width, img_height=img_height