update app.py

app.py

@@ -1,7 +1,20 @@
-from fastapi import FastAPI
+import os
+import subprocess
 
-app = FastAPI()
+if __name__ == "__main__":
+    # Ensure the /data directory exists
+    os.makedirs("/data/mlruns", exist_ok=True)
+    os.makedirs("/data", exist_ok=True)
 
-@app.get("/")
-def greet_json():
-    return {"Hello": "World!"}
+    # 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"
+    ])