Update Dockerfile

Dockerfile

@@ -29,7 +29,8 @@ WORKDIR $HOME/app
 # Copy the requirements.txt first to leverage Docker cache
 COPY --chown=user requirements.txt $HOME/app/
 RUN pip install --no-cache-dir -r requirements.txt
-# RUN pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
+RUN pip uninstall --y faiss-cpu & pip install faiss-gpu
+RUN pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
 
 # Copy the rest of the application's code to the container
 COPY --chown=user . $HOME/app