Update Dockerfile

Dockerfile

@@ -1,7 +1,7 @@
 FROM python:3.10
 
 # Install dependencies
-RUN pip install --no-cache-dir mlflow huggingface_hub psycopg2-binary
+RUN pip install --no-cache-dir mlflow huggingface_hub psycopg2-binary uvicorn
 
 # Expose port
 EXPOSE 7860
@@ -11,9 +11,10 @@ ENV HF_TOKEN=${HF_TOKEN}
 ENV MLFLOW_HF_ARTIFACT_REPO=${MLFLOW_HF_ARTIFACT_REPO}
 ENV DATABASE_URL=${DATABASE_URL}
 
-# Copy custom artifact repository
+# Copy files
 WORKDIR /app
 COPY hf_artifact_repo.py .
+COPY start_mlflow.py .
 
-# Start MLflow server (plugin auto-registers on import)
-CMD python -c "import hf_artifact_repo" && mlflow server --host 0.0.0.0 --port 7860 --backend-store-uri ${DATABASE_URL} --default-artifact-root ${MLFLOW_HF_ARTIFACT_REPO:-hf://models/omarelrayes/mlflow-artifacts} --serve-artifacts --allowed-hosts "*" --cors-allowed-origins "*"
+# Start custom MLflow server
+CMD ["python", "start_mlflow.py"]