Pharma commited on
Commit
1ef5b8b
1 Parent(s): ab0575a

Update call to Stable Diffusion space

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ img_to_text = gr.Blocks.load(name="spaces/pharma/CLIP-Interrogator")
6
  stable_diffusion = gr.Blocks.load(name="spaces/stabilityai/stable-diffusion")
7
 
8
  def get_images(prompt):
9
- gallery_dir = stable_diffusion(prompt, fn_index=2)
10
  sd_output = [os.path.join(gallery_dir, image) for image in os.listdir(gallery_dir)]
11
  return sd_output, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
12
 
 
6
  stable_diffusion = gr.Blocks.load(name="spaces/stabilityai/stable-diffusion")
7
 
8
  def get_images(prompt):
9
+ gallery_dir = stable_diffusion(prompt, "", 7.5, fn_index=2)
10
  sd_output = [os.path.join(gallery_dir, image) for image in os.listdir(gallery_dir)]
11
  return sd_output, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
12