Update app.py
Browse files
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="
|
7 |
|
8 |
def get_images(prompt):
|
9 |
-
gallery_dir = stable_diffusion(
|
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 |
|