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