Update app_parquet.py

app_parquet.py

@@ -186,13 +186,15 @@ PARQUET_DATASET_DIR.mkdir(parents=True, exist_ok=True)
 
 scheduler = ParquetScheduler(repo_id="example-space-to-dataset-parquet")
 
-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]]:
     """Generate images on 'submit' button."""
     # Generate from https://huggingface.co/spaces/stabilityai/stable-diffusion
-    out_dir = client.predict(prompt, "", 9, fn_index=1)
+    # out_dir = client.predict(prompt, "", 9, fn_index=1) # Space 'stabilityai/stable-diffusion' is paused
+    out_dir = client.predict(prompt, fn_index=1)
     with (Path(out_dir) / "captions.json").open() as f:
         paths = list(json.load(f).keys())