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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,10 +3,10 @@ import os
3
  from share_btn import community_icon_html, loading_icon_html, share_js
4
 
5
  text_gen = gr.Interface.load(name="spaces/Gustavosta/MagicPrompt-Stable-Diffusion")
6
- stable_diffusion = gr.Blocks.load(name="spaces/stabilityai/stable-diffusion")
7
 
8
  def get_images(prompt):
9
- gallery_dir = stable_diffusion(text=prompt, negative_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
 
 
3
  from share_btn import community_icon_html, loading_icon_html, share_js
4
 
5
  text_gen = gr.Interface.load(name="spaces/Gustavosta/MagicPrompt-Stable-Diffusion")
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