add cuda

Dockerfile

@@ -14,6 +14,8 @@ COPY pyproject.toml poetry.lock /app/
 
 # Install poetry
 RUN pip install poetry
+RUN pip install torch==1.10.0+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html
+
 
 # Set environment variable to create virtualenv within the project directory
 ENV POETRY_VIRTUALENVS_IN_PROJECT=true

pyproject.toml

@@ -8,7 +8,7 @@ readme = "README.md"
 
 [tool.poetry.dependencies]
 python = "^3.10"
-torch = {version = "1.10.0+cpu", url = "https://download.pytorch.org/whl/cpu/torch_stable.html"}
+torch = "1.10.0"
 farm-haystack = {extras = ["inference"], version = "^1.20.1"}
 chainlit = "^0.7.0"
 openai = "^0.28.0"