fix output dir

app_base.py

@@ -30,7 +30,6 @@ def create_demo() -> gr.Blocks:
         num_steps: int,
         start_step: int,
         guidance_scale: float,
-        segment_size: int,
         generate_size: int,
         upscale_prompt: str,
         upscale_start_size: int = 256,
@@ -73,7 +72,6 @@ def create_demo() -> gr.Blocks:
             upscale_steps=upscale_steps,
         )
         run_task_time, time_cost_str = get_time_cost(run_task_time, time_cost_str)
-        enhanced_image = enhanced_image.resize((segment_size, segment_size))
 
         return enhanced_image, res_image, time_cost_str
 
@@ -135,7 +133,7 @@ def create_demo() -> gr.Blocks:
         ).success(
             fn=image_to_image,
             inputs=[origin_area_image, input_image_prompt, edit_prompt,seed,w1, num_steps, start_step,
-                guidance_scale, segment_size, generate_size, upscale_prompt, upscale_start_size, upscale_steps,
+                guidance_scale, generate_size, upscale_prompt, upscale_start_size, upscale_steps,
                 pre_upscale, pre_upscale_start_size, pre_upscale_steps],
             outputs=[enhanced_image, generated_image, generated_cost],
         ).success(

segment_utils.py

@@ -1,6 +1,7 @@
 import numpy as np
 import mediapipe as mp
 import uuid
+import os
 
 from PIL import Image
 from mediapipe.tasks import python
@@ -29,7 +30,13 @@ def restore_result(croper, category, generated_image):
     # else:
     #     extension = 'jpg'
 
-    path = f"output/{uuid.uuid4()}.{extension}"
+    tmpPrefix = "/tmp/gradio/"
+
+    targetDir = f"{tmpPrefix}output/"
+    if not os.path.exists(targetDir):
+        os.makedirs(targetDir)
+
+    path = f"{targetDir}{uuid.uuid4()}.{extension}"
     restored_image.save(path, quality=95)
 
     return restored_image, path