Update Docker

Docker

@@ -1,24 +1,24 @@
 # Use a lightweight Python image
 FROM python:3.9-slim
 
+# Set environment variables
+ENV PYTHONUNBUFFERED=1 \
+    SHELL=/bin/bash
+
 # Set the working directory
 WORKDIR /app
 
 # Install MLflow
 RUN pip install mlflow
 
-# Expose the default MLflow port
-EXPOSE 7860
-
+# Copy the start_server.sh script into the container
 COPY start_server.sh .
 
+# Ensure the script is executable
 RUN chmod +x start_server.sh
 
-# Ensure the /data directory exists
-RUN mkdir -p /data/mlruns
-
-# Set environment variables
-ENV MLFLOW_TRACKING_URI=file:///data/mlruns
+# Expose the default MLflow port
+EXPOSE 7860
 
-# Command to run the MLflow server
+# Command to run the start_server.sh script
 CMD ["./start_server.sh"]