fffiloni commited on
Commit
634ecf0
·
1 Parent(s): c33913d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -12,10 +12,11 @@ Stable Diffusion 2 uses OpenCLIP ViT-H model trained on LAION dataset so it know
12
  </p>
13
  """
14
 
15
- stable_diffusion = gr.load(name="spaces/runwayml/stable-diffusion-v1-5")
16
  clip_interrogator_2 = Client("https://fffiloni-clip-interrogator-2.hf.space/")
17
 
18
  def get_images(prompt):
 
19
  gallery_dir = stable_diffusion(prompt, fn_index=2)
20
  img_results = [os.path.join(gallery_dir, img) for img in os.listdir(gallery_dir)]
21
  return img_results[0]
 
12
  </p>
13
  """
14
 
15
+ stable_diffusion = Client("https://runwayml-stable-diffusion-v1-5.hf.space/")
16
  clip_interrogator_2 = Client("https://fffiloni-clip-interrogator-2.hf.space/")
17
 
18
  def get_images(prompt):
19
+ print("Calling SD")
20
  gallery_dir = stable_diffusion(prompt, fn_index=2)
21
  img_results = [os.path.join(gallery_dir, img) for img in os.listdir(gallery_dir)]
22
  return img_results[0]