Update app.py

app.py

@@ -39,7 +39,8 @@ def main():
         gr.Markdown("# Instruction Tuning with Unsloth")
     
         ##### Model Inputs #####
-    
+        gr.Markdown("### Model Inputs")
+        
         # Select Model
         modelnames = conf['model']['choices']
         model_name = gr.Dropdown(label="Supported Models", 
@@ -67,6 +68,7 @@ def main():
                               dataset_choice, 
                               dataset_upload)
 
+        gr.Markdown("### Model Parameter Selection")
         # Parameters
         max_seq_length = gr.Textbox(label="Maximum sequence length", 
                                      value=conf['model']['general']["max_seq_length"])
@@ -96,6 +98,7 @@ def main():
     
         # Setup button
         tune_btn = gr.Button("Start Fine Tuning")
+        gr.Markdown("### Model Progress")
         # Text output (for now)
         output = gr.Textbox(label="Output") 
         # Execute button