Update app_parquet.py

app_parquet.py

@@ -187,14 +187,14 @@ PARQUET_DATASET_DIR.mkdir(parents=True, exist_ok=True)
 scheduler = ParquetScheduler(repo_id="example-space-to-dataset-parquet")
 
 # client = Client("stabilityai/stable-diffusion") # Space is paused
-client = Client("runwayml/stable-diffusion-v1-5")
+client = Client("black-forest-labs/FLUX.1-dev")
 
 
 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) # Space 'stabilityai/stable-diffusion' is paused
-    out_dir = client.predict(prompt, fn_index=1)
+    out_dir = client.predict(prompt, api_name="/infer")
     with (Path(out_dir) / "captions.json").open() as f:
         paths = list(json.load(f).keys())