anzorq commited on
Commit
2355e4a
1 Parent(s): 5a53a51

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -98,7 +98,7 @@ def generate_txt2img(prompt):
98
 
99
  stable_diffusion = gr.Blocks.load(name="spaces/runwayml/stable-diffusion-v1-5")
100
  prompt = f"“a 3d rendering of {prompt}, chair imitating an avocado, full view, white background"
101
- gallery_dir = stable_diffusion(prompt, fn_index=2)[0]
102
  imgs = [os.path.join(gallery_dir, img) for img in os.listdir(gallery_dir)]
103
 
104
  return imgs[1], gr.update(visible=True)
 
98
 
99
  stable_diffusion = gr.Blocks.load(name="spaces/runwayml/stable-diffusion-v1-5")
100
  prompt = f"“a 3d rendering of {prompt}, chair imitating an avocado, full view, white background"
101
+ gallery_dir = stable_diffusion(prompt, fn_index=2)
102
  imgs = [os.path.join(gallery_dir, img) for img in os.listdir(gallery_dir)]
103
 
104
  return imgs[1], gr.update(visible=True)