GRADIO_TEMP_DIR fix (#3)

- GRADIO_TEMP_DIR fix (0deab391eafa1360e583e4a3a40cb5574010d578)


Co-authored-by: Morpheus <[email protected]>

app.py

@@ -7,6 +7,7 @@ import gc
 import cv2
 import torch
 import random
+import tempfile
 import numpy as np
 from PIL import Image
 from diffusers import ControlNetModel, StableDiffusionXLControlNetPipeline
@@ -163,7 +164,7 @@ def create_image(image_pil,
                  target="Load only style blocks",
                  neg_content_prompt=None,
                  neg_content_scale=0,
-                 progress=gr.Progress(track_tqdm=True)
+#                 progress=gr.Progress(track_tqdm=True)
                 ):
 
     if target =="Load original IP-Adapter":
@@ -215,7 +216,7 @@ def create_image(image_pil,
                                 controlnet_conditioning_scale=float(control_scale),
                                 )
     
-    gradio_temp_dir = os.environ['GRADIO_TEMP_DIR']
+    gradio_temp_dir = os.environ.get('GRADIO_TEMP_DIR', tempfile.gettempdir())
     temp_file_path = os.path.join(gradio_temp_dir, "image.png")
     images[0].save(temp_file_path, format="PNG")
     print(f"Image saved in: {temp_file_path}")