Update app.py

app.py

@@ -108,42 +108,47 @@ def submit_weights(model, repository, model_out_name, token):
 def main():
     with gr.Blocks() as demo:
 
-
         with gr.Tabs():
             with gr.TabItem("About"):
+                # About page!!
                 gr.Markdown(get_files.load_markdown_file("README.md"))
-                
-            with gr.TabItem("Train Model"):
-                ##### Title Block #####
-                gr.Markdown("# SLM Instruction Tuning with Unsloth")
-            
-                ##### Initial Model Inputs #####
-                gr.Markdown("### Model Inputs")
-                
+
+            with gr.TabItem("Basic Setup"):
+                gr.Markdown("# Select Model and Input details")
                 # Select Model
                 modelnames = conf['model']['choices']
                 model_name = gr.Dropdown(label="Supported Models", 
                                          choices=modelnames, 
                                          value=modelnames[0])
-                # Prompt template
-                inject_prompt = gr.Textbox(label="Prompt Template", 
-                                             value=prompt_template())
-                # Dataset choice
+                # Select Generic Model parameters
+                repository = gr.Textbox(label="Repository Name",
+                                        value=conf['model']['general']["repository"])   
+                model_out_name = gr.Textbox(label="Model Output Name",
+                                        value=conf['model']['general']["model_name"])
+                hf_token = gr.Textbox(label="Huggingface Token",
+                                      type='password',
+                                      value='')
+                
+
+            with gr.TabItem("Upload Data"):
+                # Toggle dataset load types
+                gr.Markdown("# Dataset Selection and Upload")
+                
                 dataset_choice = gr.Radio(label="Choose Dataset", 
                                           choices=["Hugging Face Hub Dataset", "Upload Your Own"], 
                                           value="Hugging Face Hub Dataset")
-        
                 dataset_predefined = gr.Textbox(label="Hugging Face Hub Training Dataset", 
                                                 value='yahma/alpaca-cleaned', 
                                                 visible=True)
-                
                 dataset_predefined_load = gr.Button("Upload Dataset (.csv, .jsonl, or .txt)")
         
                 dataset_uploaded_load = gr.UploadButton(label="Upload Dataset (.csv, .jsonl, or .txt)", 
                                                  file_types=[".csv",".jsonl", ".txt"], 
                                                  visible=False)
+                # Safety output to show if upload succeeded.
                 data_snippet = gr.Markdown()
-                
+
+                # Visibility toggler
                 dataset_choice.change(update_visibility.textbox_vis, 
                                       dataset_choice, 
                                       dataset_predefined)
@@ -153,8 +158,10 @@ def main():
                 dataset_choice.change(update_visibility.textbox_button_vis,
                                       dataset_choice,
                                       dataset_predefined_load)
-                
-                # Dataset button
+                # Prompt template
+                inject_prompt = gr.Textbox(label="Prompt Template", 
+                                             value=prompt_template())
+                # Dataset buttons
                 dataset_predefined_load.click(fn=get_files.predefined_dataset,
                                           inputs=dataset_predefined,
                                           outputs=data_snippet)
@@ -162,13 +169,12 @@ def main():
                 dataset_uploaded_load.click(fn=get_files.uploaded_dataset,
                                          inputs=dataset_uploaded_load,
                                          outputs=data_snippet)
-        
-        
-        
-                
-        
+         
+            
+            with gr.TabItem("Train Model"):
                 ##### Model Parameter Inputs #####
-                gr.Markdown("### Model Parameter Selection")
+                gr.Markdown("# Model Parameter Selection")
+
                 # Parameters
                 data_field = gr.Textbox(label="Dataset Training Field Name",
                                         value=conf['model']['general']["dataset_text_field"])
@@ -179,11 +185,7 @@ def main():
                 num_epochs = gr.Textbox(label="Training Epochs",
                                         value=conf['model']['general']["num_train_epochs"])
                 max_steps = gr.Textbox(label="Maximum steps",
-                                        value=conf['model']['general']["max_steps"])   
-                repository = gr.Textbox(label="Repository Name",
-                                        value=conf['model']['general']["repository"])   
-                model_out_name = gr.Textbox(label="Model Output Name",
-                                        value=conf['model']['general']["model_name"])   
+                                        value=conf['model']['general']["max_steps"])    
         
                 # Hyperparameters (allow selection, but hide in accordion.)
                 with gr.Accordion("Advanced Tuning", open=False):