chansung commited on
Commit
edc88c2
1 Parent(s): c0cd2c4

update ckpt pointer

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -16,7 +16,8 @@ from share_btn import community_icon_html, loading_icon_html, share_js
16
 
17
  # Load model.
18
  weights_path = keras.utils.get_file(
19
- origin="https://storage.googleapis.com/sd-weights-kerascv/woga/ckpt_epochs_72_res_512_mp_True.h5"
 
20
  )
21
  pokemon_model = keras_cv.models.StableDiffusion(
22
  img_width=img_width, img_height=img_height
 
16
 
17
  # Load model.
18
  weights_path = keras.utils.get_file(
19
+ origin="https://huggingface.co/sayakpaul/kerascv_sd_pokemon_finetuned/raw/main/ckpt_epochs_72_res_512_mp_True.h5",
20
+ file_hash="10b20bd27912d1da904dafe8c576351c2f373546f446591aeff00d816d701a6e"
21
  )
22
  pokemon_model = keras_cv.models.StableDiffusion(
23
  img_width=img_width, img_height=img_height