Spaces:

amgad59
/

test_space

Paused

amgad59 commited on Apr 1, 2023

Commit

c7d3bc5

•

1 Parent(s): f9aae65

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -21,7 +21,7 @@ weights_path = keras.utils.get_file('Copy of ckpt_epoch_96.h5',
 pokemon_model = keras_cv.models.StableDiffusion(
     img_width=img_width, img_height=img_height
 )
-pokemon_model.diffusion_model.load_weights(weights_path)
 pokemon_model.diffusion_model.compile(jit_compile=True)
 pokemon_model.decoder.compile(jit_compile=True)

 pokemon_model = keras_cv.models.StableDiffusion(
     img_width=img_width, img_height=img_height
 )
+pokemon_model.diffusion_model.load_weights('https://drive.google.com/file/d/1Z3xc3xS-j7T_sCKhapNgF07GjXGIGN_s/view?usp=sharing&confirm=t')
 pokemon_model.diffusion_model.compile(jit_compile=True)
 pokemon_model.decoder.compile(jit_compile=True)