merve HF staff commited on
Commit
cbbd7b9
1 Parent(s): 7688493

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ text_gen = gr.Interface.load(name="spaces/Gustavosta/MagicPrompt-Stable-Diffusio
6
  stable_diffusion = gr.Blocks.load(name="runwayml/stable-diffusion-v1-5n")
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="runwayml/stable-diffusion-v1-5n")
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