Update flux_train_ui.py

flux_train_ui.py

@@ -23,13 +23,14 @@ import uuid
 from slugify import slugify
 import gradio as gr  # Assuming gr is from gradio for error/warning handling
 
+os.makedirs("tmp", exist_ok=True)
 # Configure logging
 logging.basicConfig(
     level=logging.DEBUG,
     format='%(asctime)s - %(levelname)s - %(message)s',
     handlers=[
         logging.StreamHandler(),  # Output to console
-        logging.FileHandler('training.log')  # Save logs to a file
+        logging.FileHandler('tmp/training.log')  # Save logs to a file
     ]
 )
 logger = logging.getLogger(__name__)