Updates to the docker file for jupyterlab

docker/Dockerfile

@@ -1,4 +1,5 @@
-FROM runpod/base:0.6.2-cuda12.1.0
+FROM runpod/base:0.6.2-cuda12.2.0
+
 LABEL authors="jaret"
 
 # Install dependencies
@@ -17,5 +18,10 @@ RUN python -m pip install -r requirements.txt
 
 RUN apt-get install -y tmux nvtop htop
 
+RUN pip install jupyterlab
+
+# symlink app to workspace
+RUN ln -s /app/ai-toolkit /workspace/ai-toolkit
+
 WORKDIR /
 CMD ["/start.sh"]