Update Dockerfile

Dockerfile

@@ -6,6 +6,7 @@ COPY ./requirements.txt /code/requirements.txt
 COPY ./pre-requirements.txt /code/pre-requirements.txt
 
 RUN pip install --no-cache-dir --upgrade -r /code/pre-requirements.txt
+RUN pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu117
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
 # Set up a new user named "user" with user ID 1000