Update Dockerfile

Dockerfile

@@ -1,7 +1,7 @@
 FROM python:3.10
 
 # Install dependencies
-RUN pip install --no-cache-dir mlflow huggingface_hub psycopg2-binary uvicorn
+RUN pip install --no-cache-dir mlflow huggingface_hub psycopg2-binary
 
 # Expose port
 EXPOSE 7860
@@ -10,10 +10,12 @@ EXPOSE 7860
 ENV HF_TOKEN=${HF_TOKEN}
 ENV MLFLOW_HF_ARTIFACT_REPO=${MLFLOW_HF_ARTIFACT_REPO}
 ENV DATABASE_URL=${DATABASE_URL}
+ENV PYTHONPATH=/app:${PYTHONPATH}
 
 # Copy files
 WORKDIR /app
 COPY hf_artifact_repo.py .
+COPY sitecustomize.py .
 COPY start_mlflow.py .
 
 # Start custom MLflow server