change to SD 2.1
Browse files
app.py
CHANGED
|
@@ -10,7 +10,7 @@ import os
|
|
| 10 |
#from diffusers import StableDiffusionPipeline
|
| 11 |
|
| 12 |
whisper = gr.Blocks.load(name="spaces/sanchit-gandhi/whisper-large-v2")
|
| 13 |
-
stable_diffusion = gr.Blocks.load(name="spaces/
|
| 14 |
### ββββββββββββββββββββββββββββββββββββββββ
|
| 15 |
|
| 16 |
title="Whisper to Stable Diffusion"
|
|
@@ -27,7 +27,7 @@ title="Whisper to Stable Diffusion"
|
|
| 27 |
### ββββββββββββββββββββββββββββββββββββββββ
|
| 28 |
|
| 29 |
def get_images(prompt):
|
| 30 |
-
gallery_dir = stable_diffusion(prompt, fn_index=2)
|
| 31 |
return [os.path.join(gallery_dir, img) for img in os.listdir(gallery_dir)]
|
| 32 |
|
| 33 |
|
|
|
|
| 10 |
#from diffusers import StableDiffusionPipeline
|
| 11 |
|
| 12 |
whisper = gr.Blocks.load(name="spaces/sanchit-gandhi/whisper-large-v2")
|
| 13 |
+
stable_diffusion = gr.Blocks.load(name="spaces/stabilityai/stable-diffusion")
|
| 14 |
### ββββββββββββββββββββββββββββββββββββββββ
|
| 15 |
|
| 16 |
title="Whisper to Stable Diffusion"
|
|
|
|
| 27 |
### ββββββββββββββββββββββββββββββββββββββββ
|
| 28 |
|
| 29 |
def get_images(prompt):
|
| 30 |
+
gallery_dir = stable_diffusion(prompt,"", 9, fn_index=2)
|
| 31 |
return [os.path.join(gallery_dir, img) for img in os.listdir(gallery_dir)]
|
| 32 |
|
| 33 |
|