Yusin commited on
Commit
d25de88
β€’
1 Parent(s): 585f8a8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
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
- #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
  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