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 |
|