Update app.py

app.py

@@ -4,6 +4,11 @@ import os
 from huggingface_hub import snapshot_download
 from flux_train import build_job
 from toolkit.job import run_job
+import sys
+sys.path.append("/app/ai-toolkit")  # Tell Python to look here
+
+from toolkit.job import run_job
+
 
 app = FastAPI()