Spaces:

amgad59
/

test_space

Paused

amgad59 commited on Apr 1, 2023

Commit

a080da2

1 Parent(s): 8f2e8cd

Update app.py

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://huggingface.co/sayakpaul/kerascv_sd_pokemon_finetuned/resolve/main/ckpt_epochs_577_res_256_mp_False.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/andrew27/kerascv_clothing_finetuned/resolve/main/ckpt_epoch_96.h5",
+    file_hash="4b4348297aa9853ff9dc4da7f52dcb240210564400f164e5155e5f4dc1866626"
 )
 pokemon_model = keras_cv.models.StableDiffusion(
     img_width=img_width, img_height=img_height