chore: Update Dockerfile to install nvidia-container-toolkit for GPU support

Dockerfile

@@ -37,6 +37,11 @@ ENV PATH="/home/chatbot/.local/bin:${PATH}"
 #    && sudo cp /var/cudnn-local-repo-ubuntu2204-9.1.1/cudnn-*-keyring.gpg /usr/share/keyrings/ \
 #    && sudo apt-get update && sudo apt-get -y install cudnn 
 
+RUN curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg \
+  && curl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list | \
+    sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | \
+    sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list && sudo apt-get update && sudo apt-get install -y nvidia-container-toolkit
+
 # Create a working directory
 WORKDIR /app
 

start_chatbot.sh

@@ -1,2 +1,5 @@
 echo "#### BOT BOOT ####"
+
+cd src
+
 uvicorn main:app