Switch to 'runwayml/stable-diffusion-v1-5' Space (#3)

- Switch to 'runwayml/stable-diffusion-v1-5' Space (db06a63d24256c9c8f9c9720db396c59f2b2439f)


Co-authored-by: Lucain Pouget <[email protected]>

app.py

@@ -34,13 +34,15 @@ scheduler = ParquetScheduler(
     token=HF_TOKEN,
 )
 
-client = Client("stabilityai/stable-diffusion")
+# client = Client("stabilityai/stable-diffusion") # Space is paused
+client = Client("runwayml/stable-diffusion-v1-5")
 
 
 def generate(prompt: str) -> tuple[str, list[str]]:
     negative_prompt = ""
     guidance_scale = 9.0
-    out_dir = client.predict(prompt, negative_prompt, guidance_scale, fn_index=1)
+    # out_dir = client.predict(prompt, negative_prompt, guidance_scale, fn_index=1) # Space 'stabilityai/stable-diffusion' is paused
+    out_dir = client.predict(prompt, fn_index=1)
 
     config = {
         "prompt": prompt,