update torch install

swwho96 · swwho96 · commit c8e4529f611f · 2025-12-25T23:39:30.000+09:00
diff --git a/Dockerfile b/Dockerfile
@@ -2,6 +2,8 @@ FROM python:3.11-slim
 
 WORKDIR /codive-ai
 
+RUN pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
+
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
diff --git a/requirements.txt b/requirements.txt
@@ -4,7 +4,6 @@ gunicorn
 python-multipart
 aioboto3
 rembg[cpu]
-torch
 open-clip-torch
 pillow
 numpy
