.

Dockerfile

@@ -1,11 +1,20 @@
 FROM pytorch/pytorch:latest
 
-# Install Jupyter Notebook
+# Install JupyterLab
 RUN pip install jupyterlab
 
 # Install additional packages
-RUN pip install tensorflow
+RUN pip install plotly
 
 # Set environment variables, etc.
 #ENV MY_ENV_VAR=myvalue
 
+# Set the working directory
+WORKDIR /workspace
+
+# Expose the port for JupyterLab
+EXPOSE 8888
+
+# Command to run JupyterLab
+CMD ["jupyter-lab", "--ip=0.0.0.0", "--port=8888", "--no-browser", "--allow-root", "--NotebookApp.token=''"]
+