Update Docker

Docker

@@ -10,6 +10,10 @@ RUN pip install mlflow
 # Expose the default MLflow port
 EXPOSE 7860
 
+COPY start_server.sh .
+
+RUN chmod +x start_server.sh
+
 # Ensure the /data directory exists
 RUN mkdir -p /data/mlruns
 
@@ -17,6 +21,4 @@ RUN mkdir -p /data/mlruns
 ENV MLFLOW_TRACKING_URI=file:///data/mlruns
 
 # Command to run the MLflow server
-CMD ["mlflow", "server", "--host", "0.0.0.0", "--port", "7860", \
-     "--backend-store-uri", "sqlite:////data/mlflow.db", \
-     "--default-artifact-root", "/data/mlruns"]
+CMD ["./app.sh"]