Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -4,13 +4,14 @@ import os
|
|
4 |
|
5 |
#from diffusers import StableDiffusionPipeline
|
6 |
whisper = gr.Interface.load(name="spaces/sanchit-gandhi/whisper-large-v2")
|
7 |
-
stable_diffusion = gr.Blocks.load(name="spaces/stabilityai/stable-diffusion")
|
8 |
-
|
9 |
### ββββββββββββββββββββββββββββββββββββββββ
|
10 |
title="Talking to Stable Diffusion"
|
11 |
### ββββββββββββββββββββββββββββββββββββββββ
|
12 |
def get_images(prompt):
|
13 |
-
gallery_dir = stable_diffusion(prompt, None, None, fn_index=2)
|
|
|
14 |
return [os.path.join(gallery_dir, img) for img in os.listdir(gallery_dir)]
|
15 |
|
16 |
|
|
|
4 |
|
5 |
#from diffusers import StableDiffusionPipeline
|
6 |
whisper = gr.Interface.load(name="spaces/sanchit-gandhi/whisper-large-v2")
|
7 |
+
#stable_diffusion = gr.Blocks.load(name="spaces/stabilityai/stable-diffusion")
|
8 |
+
stable_diffusion = gr.Blocks.load(name="spaces/runwayml/stable-diffusion-v1-5")
|
9 |
### ββββββββββββββββββββββββββββββββββββββββ
|
10 |
title="Talking to Stable Diffusion"
|
11 |
### ββββββββββββββββββββββββββββββββββββββββ
|
12 |
def get_images(prompt):
|
13 |
+
#gallery_dir = stable_diffusion(prompt, None, None, fn_index=2)
|
14 |
+
gallery_dir = stable_diffusion(prompt, fn_index=2)
|
15 |
return [os.path.join(gallery_dir, img) for img in os.listdir(gallery_dir)]
|
16 |
|
17 |
|