Update app.py

app.py

@@ -75,12 +75,14 @@ else:
 
 
 def _run_pipe(image, prompt, seed, steps):
+    print(f"[_run_pipe] start cuda_avail={torch.cuda.is_available()}", flush=True)
     gc.collect()
     if torch.cuda.is_available():
         torch.cuda.empty_cache()
     width, height = _round_dims(image)
+    print(f"[_run_pipe] dims w={width} h={height} steps={steps}", flush=True)
     generator = torch.Generator(device=DEVICE).manual_seed(seed)
-    return PIPE(
+    out = PIPE(
         image=[image],
         prompt=prompt,
         negative_prompt=NEGATIVE_PROMPT,
@@ -90,6 +92,8 @@ def _run_pipe(image, prompt, seed, steps):
         true_cfg_scale=1.0,
         generator=generator,
     ).images[0]
+    print(f"[_run_pipe] done size={out.size}", flush=True)
+    return out
 
 
 # --- Server -------------------------------------------------------------------
@@ -97,18 +101,23 @@ server = gr.Server()
 HOME = Path(__file__).parent
 
 
-@server.api(name="edit_image", concurrency_limit=1)
+@server.api(name="edit_image")
 def edit_image(image: FileData, prompt: str) -> dict:
     """Edit an image guided by a text prompt using FireRed-Image-Edit 1.1."""
+    print(f"[edit_image] received prompt={prompt!r} path={image.get('path')}", flush=True)
     if not prompt or not prompt.strip():
         return {"error": "Please enter an edit prompt."}
     src = Image.open(image["path"]).convert("RGB")
+    print(f"[edit_image] image opened size={src.size}", flush=True)
     seed = random.randint(0, MAX_SEED)
+    print(f"[edit_image] calling _edit seed={seed}", flush=True)
     result = _edit(src, prompt.strip(), seed, steps=4)
+    print(f"[edit_image] _edit returned size={result.size}", flush=True)
 
     fd, out_path = tempfile.mkstemp(suffix=".png")
     os.close(fd)
     result.save(out_path)
+    print(f"[edit_image] saved to {out_path}", flush=True)
     return {"image": FileData(path=out_path), "seed": seed}
 
 
@@ -118,4 +127,4 @@ async def homepage():
 
 
 if __name__ == "__main__":
-    server.launch(mcp_server=True, show_error=True)
+    server.launch(show_error=True)