Add torchvision for CPU Space

Dockerfile

@@ -9,7 +9,7 @@ ENV PYTHONDONTWRITEBYTECODE=1 \
     DEPTH_PRO_DTYPE=float32 \
     DEPTH_PRO_EAGER_LOAD=false \
     MAX_IMAGE_BYTES=16777216 \
-    HF_HUB_ENABLE_HF_TRANSFER=1
+    HF_XET_HIGH_PERFORMANCE=1
 
 WORKDIR /app
 
@@ -17,7 +17,7 @@ RUN useradd -m -u 1000 appuser
 
 COPY requirements.txt .
 RUN python -m pip install --upgrade pip \
-    && python -m pip install --index-url https://download.pytorch.org/whl/cpu torch==2.5.1 \
+    && python -m pip install --index-url https://download.pytorch.org/whl/cpu torch==2.5.1 torchvision==0.20.1 \
     && python -m pip install -r requirements.txt
 
 COPY main.py .

Dockerfile.gpu

@@ -9,7 +9,7 @@ ENV PYTHONDONTWRITEBYTECODE=1 \
     DEPTH_PRO_DTYPE=auto \
     DEPTH_PRO_EAGER_LOAD=true \
     MAX_IMAGE_BYTES=16777216 \
-    HF_HUB_ENABLE_HF_TRANSFER=1
+    HF_XET_HIGH_PERFORMANCE=1
 
 WORKDIR /app
 
@@ -17,6 +17,7 @@ RUN useradd -m -u 1000 appuser
 
 COPY requirements.txt .
 RUN python -m pip install --upgrade pip \
+    && python -m pip install --index-url https://download.pytorch.org/whl/cu124 torchvision==0.20.1 \
     && python -m pip install -r requirements.txt
 
 COPY main.py .

requirements.txt

@@ -5,5 +5,6 @@ numpy>=1.26.0
 pillow>=10.4.0
 safetensors>=0.4.5
 torch>=2.4.0
+torchvision>=0.19.0
 transformers>=4.50.0
 uvicorn[standard]>=0.30.0