Update app.py

app.py

@@ -1,9 +1,12 @@
 import os
 import sys
+
+# Environment variables
 os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
 os.environ['GRADIO_ANALYTICS_ENABLED'] = '0'
-sys.path.insert(0, os.getcwd())
-sys.path.append(os.path.join(os.path.dirname(__file__), 'sd-scripts'))
+
+# No need to adjust sys.path if using proper module imports
+
 import subprocess
 import gradio as gr
 from PIL import Image
@@ -16,13 +19,15 @@ from slugify import slugify
 from transformers import AutoProcessor, AutoModelForCausalLM
 from gradio_logsview import LogsView, LogsViewRunner
 from huggingface_hub import hf_hub_download, HfApi
-from library import flux_train_utils, huggingface_util
+from fluxgym_main.library import flux_train_utils, huggingface_util
 from argparse import Namespace
-import train_network
+from fluxgym_main import train_network
 import toml
 import re
+
 MAX_IMAGES = 150
 
+
 with open('models.yaml', 'r') as file:
     models = yaml.safe_load(file)