Update app.py

app.py

@@ -1,5 +1,6 @@
 import os
 import traceback
+import tempfile
 from typing import Optional
 
 import gradio as gr
@@ -16,7 +17,7 @@ def run_image_to_image(img: Image.Image, prompt: str, model: str) -> Image.Image
     client = InferenceClient(token=token)
 
     # Convert PIL image to bytes
-    with gr.processing_utils.TempFile(suffix=".png") as tmp:
+    with tempfile.NamedTemporaryFile(suffix=".png", delete=True) as tmp:
         img.save(tmp.name, format="PNG")
         with open(tmp.name, "rb") as f:
             input_bytes = f.read()
@@ -81,7 +82,7 @@ with gr.Blocks(title="图像编辑工具", css=".gradio-container {max-width: 98
     def _run_and_pack(img: Image.Image, p: str, m: str):
         result_img = run_image_to_image(img, p, m)
         # 保存为临时文件以供下载
-        with gr.processing_utils.TempFile(suffix=".png") as tmp:
+        with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp:
             result_img.save(tmp.name)
             return result_img, tmp.name