Spaces:

amgad59
/

test_space

Paused

amgad59 commited on Apr 1, 2023

Commit

4a147ae

•

1 Parent(s): 3b6b53d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,8 +15,8 @@ from constants import css, examples, img_height, img_width, num_images_to_gen
 from share_btn import community_icon_html, loading_icon_html, share_js
 # Load model.
-weights_path = keras.utils.get_file('Copy of ckpt_epoch_96.h5',
-    origin="https://drive.google.com/file/d/1Z3xc3xS-j7T_sCKhapNgF07GjXGIGN_s/view?usp=sharing&confirm=t"
 )
 pokemon_model = keras_cv.models.StableDiffusion(
     img_width=img_width, img_height=img_height

 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