Switch to 'runwayml/stable-diffusion-v1-5' Space

#3
by Wauplin HF Staff - opened
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -34,13 +34,15 @@ scheduler = ParquetScheduler(
34
  token=HF_TOKEN,
35
  )
36
 
37
- client = Client("stabilityai/stable-diffusion")
 
38
 
39
 
40
  def generate(prompt: str) -> tuple[str, list[str]]:
41
  negative_prompt = ""
42
  guidance_scale = 9.0
43
- out_dir = client.predict(prompt, negative_prompt, guidance_scale, fn_index=1)
 
44
 
45
  config = {
46
  "prompt": prompt,
 
34
  token=HF_TOKEN,
35
  )
36
 
37
+ # client = Client("stabilityai/stable-diffusion") # Space is paused
38
+ client = Client("runwayml/stable-diffusion-v1-5")
39
 
40
 
41
  def generate(prompt: str) -> tuple[str, list[str]]:
42
  negative_prompt = ""
43
  guidance_scale = 9.0
44
+ # out_dir = client.predict(prompt, negative_prompt, guidance_scale, fn_index=1) # Space 'stabilityai/stable-diffusion' is paused
45
+ out_dir = client.predict(prompt, fn_index=1)
46
 
47
  config = {
48
  "prompt": prompt,