Update app.py

app.py

@@ -1,5 +1,4 @@
 #!/usr/bin/env python
-
 from __future__ import annotations
 import os
 import gradio as gr
@@ -10,15 +9,21 @@ from app_upload import create_upload_demo
 from inference import InferencePipeline
 from trainer import Trainer
 
-TITLE = 'Entrenamiento de Modelos'
-
+TITLE = '# LoRA DreamBooth Training UI'
 ORIGINAL_SPACE_ID = 'lora-library/LoRA-DreamBooth-Training-UI'
 SPACE_ID = os.getenv('SPACE_ID', ORIGINAL_SPACE_ID)
+SHARED_UI_WARNING = f'''
+'''
 
 if os.getenv('SYSTEM') == 'spaces' and SPACE_ID != ORIGINAL_SPACE_ID:
     SETTINGS = f'<a href="https://huggingface.co/spaces/{SPACE_ID}/settings">Settings</a>'
 else:
     SETTINGS = 'Settings'
+CUDA_NOT_AVAILABLE_WARNING = f'''
+'''
+
+HF_TOKEN_NOT_SPECIFIED_WARNING = f'''
+'''
 
 HF_TOKEN = os.getenv('HF_TOKEN')
 
@@ -47,6 +52,9 @@ with gr.Blocks(css='style.css') as demo:
         with gr.TabItem('Test'):
             create_inference_demo(pipe, HF_TOKEN)
         with gr.TabItem('Upload'):
+            gr.Markdown('''
+            - You can use this tab to upload models later if you choose not to upload models in training time or if upload in training time failed.
+            ''')
             create_upload_demo(HF_TOKEN)
 
 demo.queue(max_size=1).launch(share=False)