Update app.py

app.py

@@ -4,9 +4,8 @@ from typing import Union
 is_spaces = True if os.environ.get("SPACE_ID") else False
 
 if is_spaces:
-    subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
     import spaces
-    
+    subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
     
 os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
 import sys
@@ -330,12 +329,6 @@ with gr.Blocks(theme=theme, css=css) as demo:
         ```bash
         git clone https://huggingface.co/spaces/flux-train/flux-lora-trainer
         cd flux-lora-trainer
-        ## Optional, start a venv environment ## 
-        python3 -m venv venv
-        source venv/bin/activate
-        # .\venv\Scripts\activate on windows
-        # install torch first
-        ## End of optional ##
         pip install requirements_local.txt
         ```
         
@@ -344,6 +337,10 @@ with gr.Blocks(theme=theme, css=css) as demo:
         git clone https://github.com/ostris/ai-toolkit.git
         cd ai-toolkit
         git submodule update --init --recursive
+        python3 -m venv venv
+        source venv/bin/activate
+        # .\venv\Scripts\activate on windows
+        # install torch first
         pip3 install torch
         pip3 install -r requirements.txt
         cd ..
@@ -354,11 +351,10 @@ with gr.Blocks(theme=theme, css=css) as demo:
         huggingface-cli login
         ```
         
-        Now you can run FLUX LoRA Ease locally with a UI by doing a simple 
+        Now you can run FLUX LoRA Ease locally by doing a simple 
         ```py
         python app.py
         ```
-        
         If you prefer command line, you can run Ostris' [AI Toolkit](https://github.com/ostris/ai-toolkit) yourself directly.
         """
         )