Update flux_train.py

flux_train.py

@@ -2,8 +2,13 @@
 
 import os
 from collections import OrderedDict
+import sys
+sys.path.append("/app/ai-toolkit")  # Tell Python to look here
+
 from toolkit.job import run_job
 
+
+
 def build_job(concept="ohamlab style", training_path="/tmp/data", lora_name="ohami_filter_autorun"):
     job = OrderedDict([
         ('job', 'extension'),