Update Dockerfile

Dockerfile

@@ -1,10 +1,20 @@
-FROM python:3.9
-
-WORKDIR /app
-
-COPY requirements.txt .
-RUN pip install -r requirements.txt
-
-COPY . .
-
-CMD ["python", "app.py"]
+# Use the official Jupyter Docker image
+FROM jupyter/base-notebook:latest
+
+# Install Jupyter Lab and additional packages
+RUN pip install --no-cache-dir \
+    jupyterlab \
+    torch \
+    tensorflow \
+    transformers \
+    datasets \
+    matplotlib \
+    seaborn \
+    scikit-learn \
+    pandas
+
+# Expose the port Jupyter Lab will run on
+EXPOSE 8888
+
+# Start Jupyter Lab
+CMD ["jupyter", "lab", "--ip=0.0.0.0", "--port=8888", "--no-browser", "--allow-root", "--NotebookApp.token=''"]