Try this header file

app.py

@@ -1,4 +1,29 @@
 import os
+import pathlib
+
+# Hugging Face Hub cache → /tmp
+os.environ["HF_HOME"]               = "/tmp/huggingface"
+os.environ["HUGGINGFACE_HUB_CACHE"] = "/tmp/huggingface"
+
+# XDG cache (for any other libraries) → /tmp
+os.environ["XDG_CACHE_HOME"]        = "/tmp/.cache"
+
+# Gradio general cache and examples cache → /tmp
+os.environ["GRADIO_CACHE_DIR"]      = "/tmp/.gradio"
+os.environ["GRADIO_EXAMPLES_CACHE"] = "/tmp/.gradio/cached_examples"
+
+# Pre-create all those directories so no one falls back to ./ .gradio or model/
+for d in [
+    "/tmp/huggingface",
+    "/tmp/.cache",
+    "/tmp/.gradio",
+    os.environ["GRADIO_EXAMPLES_CACHE"],
+]:
+    pathlib.Path(d).mkdir(parents=True, exist_ok=True)
+
+MODEL_DIR = "/tmp/model"
+pathlib.Path(MODEL_DIR).mkdir(parents=True, exist_ok=True)
+
 import time
 import torch
 from models import UNet
@@ -10,15 +35,8 @@ from huggingface_hub import hf_hub_download
 import tempfile
 import requests
 
-os.environ["HF_HOME"]             = "/tmp/huggingface"
-os.environ["HUGGINGFACE_HUB_CACHE"] = "/tmp/huggingface"
-os.environ["XDG_CACHE_HOME"]      = "/tmp/.cache"
-os.environ["GRADIO_CACHE_DIR"]    = "/tmp/.gradio"
-
-MODEL_DIR = "/tmp/model"
-os.makedirs(MODEL_DIR, exist_ok=True)
+# Path to your downloaded model
 MODEL_PATH = os.path.join(MODEL_DIR, "best_unet_model.pth")
-
 # Download model if missing
 def download_model():
     print("Starting model download at", time.strftime("%Y-%m-%d %H:%M:%S"))