changes

Dockerfile

@@ -18,7 +18,7 @@ WORKDIR /app
 COPY --chown=user ./requirements.txt requirements.txt
 
 # Install torch first
-RUN pip install --no-cache-dir torch==2.0.1+cu117
+# RUN pip install --no-cache-dir torch==2.0.1+cu117
 
 # Now install the rest of the packages
 RUN pip install --no-cache-dir --upgrade -r requirements.txt

requirements.txt

@@ -3,6 +3,7 @@ pillow
 numpy
 fastapi 
 uvicorn
+torch==2.0.1
 torchvision==0.15.2
 transformers==4.37.2
 tiktoken==0.6.0