add cuda

Dockerfile

@@ -28,9 +28,6 @@ COPY pyproject.toml poetry.lock /app/
 RUN pip install --upgrade pip
 RUN pip install poetry
 
-# Install the CPU-only version of torch
-RUN pip install torch==1.11.0+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html
-
 # Set environment variable to create a virtual environment within the project directory
 ENV POETRY_VIRTUALENVS_IN_PROJECT=true
 

pyproject.toml

@@ -8,7 +8,9 @@ readme = "README.md"
 
 [tool.poetry.dependencies]
 python = "^3.9"
-torch = "1.11.0+cpu"
+torch = [
+  {url = "https://download.pytorch.org/whl/cpu/torch-1.10.0%2Bcpu-cp39-cp39-linux_x86_64.whl", markers = "sys_platform == 'linux'"},
+]
 farm-haystack = {extras = ["inference"], version = "^1.20.1"}
 chainlit = "^0.7.0"
 openai = "^0.28.0"