asign cache to /data

app.py

@@ -7,6 +7,19 @@ import gradio as gr
 import spaces
 
 import os
+# Hugging Face 相关缓存
+os.environ["HF_HOME"] = "/data/.huggingface"
+os.environ["HF_HUB_CACHE"] = "/data/.cache/huggingface/hub"
+os.environ["HF_DATASETS_CACHE"] = "/data/.cache/huggingface/datasets"
+os.environ["TRANSFORMERS_CACHE"] = "/data/.cache/huggingface/hub"
+
+# PyTorch / Torch Hub / TorchVision / 部分 timm 权重
+os.environ["TORCH_HOME"] = "/data/.cache/torch"
+os.environ["XDG_CACHE_HOME"] = "/data/.cache"
+
+# 关键：PyTorch JIT CUDA/C++ 扩展编译缓存目录
+os.environ["TORCH_EXTENSIONS_DIR"] = "/data/.cache/torch_extensions"
+
 import uuid
 from typing import Any, List, Optional, Union
 import cv2
@@ -46,7 +59,7 @@ generated_object_map = {}
 
 ############## 3D-Fixer model
 model_dir = 'HorizonRobotics/3D-Fixer'
-local_dir = 'checkpoints/3D-Fixer'
+local_dir = "/data/checkpoints/3D-Fixer"
 snapshot_download(repo_id=model_dir, local_dir=local_dir)
 ############## 3D-Fixer model
 
@@ -192,12 +205,6 @@ def run_depth_estimation(
 ) -> Image.Image:
     rgb_image = image_prompts["image"].convert("RGB")
 
-    import torch
-    print(torch.__version__)
-    print(torch.version.cuda)
-    print(torch.__file__)
-    print(torch.cuda.is_available())
-
     from threeDFixer.datasets.utils import (
         normalize_vertices,
         project2ply