Hope this works

app.py

@@ -1,29 +1,30 @@
 import os
 import pathlib
 
-# Hugging Face Hub cache → /tmp
+# ─── Redirect Hugging Face & other caches into /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
+# ─── Redirect Gradio’s caches (general + examples) into /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 [
+# ─── Pre-create all writable dirs so nothing 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)
 
+# ─── Your model directory under /tmp ───────────────────────────────────────
 MODEL_DIR = "/tmp/model"
 pathlib.Path(MODEL_DIR).mkdir(parents=True, exist_ok=True)
+MODEL_PATH = os.path.join(MODEL_DIR, "best_unet_model.pth")
 
+# ─── Now safely import everything else ────────────────────────────────────
 import time
 import torch
 from models import UNet
@@ -35,6 +36,7 @@ from huggingface_hub import hf_hub_download
 import tempfile
 import requests
 
+
 # Path to your downloaded model
 MODEL_PATH = os.path.join(MODEL_DIR, "best_unet_model.pth")
 # Download model if missing