amgad59 commited on
Commit
8f2e8cd
1 Parent(s): 4a147ae

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ weights_path = keras.utils.get_file(
21
  pokemon_model = keras_cv.models.StableDiffusion(
22
  img_width=img_width, img_height=img_height
23
  )
24
- pokemon_model.diffusion_model.load_weights('https://drive.google.com/file/d/1Z3xc3xS-j7T_sCKhapNgF07GjXGIGN_s/view?usp=sharing&confirm=t')
25
 
26
  pokemon_model.diffusion_model.compile(jit_compile=True)
27
  pokemon_model.decoder.compile(jit_compile=True)
 
21
  pokemon_model = keras_cv.models.StableDiffusion(
22
  img_width=img_width, img_height=img_height
23
  )
24
+ pokemon_model.diffusion_model.load_weights(weights_path)
25
 
26
  pokemon_model.diffusion_model.compile(jit_compile=True)
27
  pokemon_model.decoder.compile(jit_compile=True)