Update Dockerfile

Dockerfile

@@ -38,6 +38,16 @@ COPY --chown=user . $HOME/app
 # Hugging Face Spaces will set $PORT automatically
 # Secrets (like AWS_ACCESS_KEY_ID) must be added via the Hugging Face UI
 
+# Define environment variables for MLflow and S3
+ARG BACKEND_STORE_URI
+ARG ARTIFACT_STORE_URI
+ARG AWS_ACCESS_KEY_ID
+ARG AWS_SECRET_ACCESS_KEY
+ENV MLFLOW_BACKEND_STORE_URI=$BACKEND_STORE_URI \
+    MLFLOW_DEFAULTArtifact_ROOT=$ARTIFACT_STORE_URI \
+    AWS_ACCESS_KEY_ID=$AWS_ACCESS_KEY_ID \
+    AWS_SECRET_ACCESS_KEY=$AWS_SECRET_ACCESS_KEY
+
 # Start the MLflow server
 CMD mlflow server \
     --host 0.0.0.0 \