revert back to working build

Docker

@@ -1,13 +0,0 @@
-FROM python:3.9
-
-RUN useradd -m -u 1000 user
-USER user
-ENV PATH="/home/user/.local/bin:$PATH"
-
-WORKDIR /app
-
-COPY --chown=user ./requirements.txt requirements.txt
-RUN pip install --no-cache-dir --upgrade -r requirements.txt
-
-COPY --chown=user . /app
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

Dockerfile

@@ -0,0 +1,17 @@
+FROM python:3.9-slim
+
+ENV PYTHONUNBUFFERED=1 \
+    SHELL=/bin/bash
+
+WORKDIR /app
+
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+COPY start_server.sh .
+
+RUN chmod +x start_server.sh
+
+EXPOSE 7860
+
+CMD ["./start_server.sh"]

app.py

@@ -1,20 +0,0 @@
-import os
-import subprocess
-
-if __name__ == "__main__":
-    # Ensure the /data directory exists
-    os.makedirs("/data/mlruns", exist_ok=True)
-    os.makedirs("/data", exist_ok=True)
-
-    # Optional: Set permissions (be cautious with permissions in production)
-    os.chmod("/data", 0o777)
-    os.chmod("/data/mlruns", 0o777)
-
-    # Start the MLflow server
-    subprocess.run([
-        "mlflow", "server",
-        "--backend-store-uri", "sqlite:////data/mlflow.db",
-        "--default-artifact-root", "/data/mlruns",
-        "--host", "0.0.0.0",
-        "--port", "7860"
-    ])

requirements.txt

@@ -1,2 +1 @@
-mlflow=~2.16.1
-fastapi
+mlflow=~2.16.1

start_server.sh

@@ -1,12 +1,14 @@
 #!/bin/bash
 
-# Ensure the /data directory exists and has proper permissions
+# Ensure the /data directory exists
 mkdir -p /data/mlruns
+
+# Optional: Set permissions
 chmod -R 777 /data
 
 # Start the MLflow server
 mlflow server \
-    --host 0.0.0.0 \
-    --port 7860 \
     --backend-store-uri sqlite:////data/mlflow.db \
-    --default-artifact-root /data/mlruns
+    --default-artifact-root /data/mlruns \
+    --host 0.0.0.0 \
+    --port 7860