Update

app.py

@@ -21,15 +21,6 @@ from uploader import upload
 TITLE = '# LoRA + StableDiffusion Training UI'
 DESCRIPTION = 'This is an unofficial demo for [https://github.com/cloneofsimo/lora](https://github.com/cloneofsimo/lora).'
 
-USAGE_INFO = '''You can train and download models in the "Training" tab, and test them in the "Test" tab.
-
-You can also test the pretrained models in the [original repo](https://github.com/cloneofsimo/lora).
-Models with names starting with "lora/" are the pretrained models and the ones with names starting with "results/" are your trained models.
-After training, you can press "Reload Weight List" button to load your trained model names.
-
-Note that your trained models will be deleted when the second training is started.
-'''
-
 ORIGINAL_SPACE_ID = 'hysts/LoRA-SD-training'
 SPACE_ID = os.getenv('SPACE_ID', ORIGINAL_SPACE_ID)
 SHARED_UI_WARNING = f'''# Attention - This Space doesn't work in this shared UI. You can duplicate and use it with a paid private T4 GPU.
@@ -103,6 +94,7 @@ def create_training_demo(trainer: Trainer,
                 gr.Markdown('''
                     - It will take about 8 minutes to train for 1000 steps with a T4 GPU.
                     - You may want to try a small number of steps first, like 1, to see if everything works fine in your environment.
+                    - Note that your trained models will be deleted when the second training is started. You can upload your trained model in the "Upload" tab.
                     ''')
 
         run_button = gr.Button('Start Training')
@@ -204,6 +196,8 @@ def create_inference_demo(pipe: InferencePipeline) -> gr.Blocks:
                 run_button = gr.Button('Generate')
 
                 gr.Markdown('''
+                - Models with names starting with "lora/" are the pretrained models provided in the [original repo](https://github.com/cloneofsimo/lora), and the ones with names starting with "results/" are your trained models.
+                - After training, you can press "Reload Weight List" button to load your trained model names.
                 - The pretrained models for "disney", "illust" and "pop" are trained with the concept prompt "style of sks".
                 - The pretrained model for "kiriko" is trained with the concept prompt "game character bnha". For this model, the text encoder is also trained.
                 ''')
@@ -283,7 +277,4 @@ with gr.Blocks(css='style.css') as demo:
         with gr.TabItem('Upload'):
             create_upload_demo()
 
-    with gr.Accordion('Usage', open=False):
-        gr.Markdown(USAGE_INFO)
-
 demo.queue(default_enabled=False).launch(share=False)