Update main.py

main.py

@@ -15,8 +15,15 @@ import imageio
 TMP_DIR = "/tmp/space_tmp"
 os.makedirs(TMP_DIR, exist_ok=True)
 
-# ✅ Load pipeline once to avoid reloading for every request
-pipeline = TrellisImageTo3DPipeline.from_pretrained("JeffreyXiang/TRELLIS-image-large")
+# ✅ Define a writable cache directory
+cache_dir = "/tmp/huggingface_cache"
+os.makedirs(cache_dir, exist_ok=True)
+
+# ✅ Manually specify cache directory when loading the model
+pipeline = TrellisImageTo3DPipeline.from_pretrained(
+    "JeffreyXiang/TRELLIS-image-large", 
+    cache_dir=cache_dir
+)
 pipeline.cuda()
 
 # ✅ Preload model (to prevent cold starts)