fffiloni commited on
Commit
6eb517f
β€’
1 Parent(s): fecd9f8

change to SD 2.1

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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/runwayml/stable-diffusion-v1-5")
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