GRADIO_TEMP_DIR fix

app.py

@@ -7,6 +7,7 @@ import gradio as gr
 import numpy as np
 import PIL
 import gc
+import tempfile
 #import spaces
 import torch
 from diffusers import LCMScheduler
@@ -68,6 +69,10 @@ pipe = StableDiffusionXLInstantIDPipeline.from_pretrained(
     safety_checker=None,
     feature_extractor=None,
 )
+
+num_inference_steps = 30
+guidance_scale = 5
+
 #pipe.cuda()
 if os.environ.get("MODE") == "Default":
     print("Default")
@@ -312,7 +317,7 @@ def generate_image(
         generator=generator,
     ).images
 
-    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}")