Update app.py

app.py

@@ -2,9 +2,6 @@ import subprocess
 import os
 import platform
 import sys
-from huggingface_hub import login
-
-login(token=os.getenv("TOKEN"))
 
 def run_command(command, cwd=None, shell=False):
     """Helper function to run a shell command."""
@@ -37,6 +34,10 @@ if not os.path.exists(clone_path):
     # Step 7: Install dependencies from requirements.txt
     run_command([sys.executable, '-m', 'pip', 'install', '-r', 'requirements.txt'], cwd=repo_dir)
 
+    from huggingface_hub import login
+
+    login(token=os.getenv("TOKEN"))
+
     # Run the flux_train_ui.py file
     run_command([sys.executable, 'flux_train_ui.py'], cwd=repo_dir)