Update app.py

app.py

@@ -29,43 +29,60 @@ def greet(model_name, inject_prompt, dataset):
     return f"Hello!! Using model: {model_name} with template: {inject_prompt}"
 
 ##################################### App UI #######################################
-with gr.Blocks() as demo:
 
-    ##### Title Block #####
-    gr.Markdown("# Instruction Tuning with Unsloth")
-
-    ##### Model Inputs #####
-
-    # 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
-    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 Dataset", value='yahma/alpaca-cleaned', visible=True)
-    dataset_upload = gr.UploadButton(label="Upload Dataset (csv, jsonl, or txt)", file_types=[".csv",".jsonl", ".txt"], visible=False)
-    dataset_choice.change(textbox_visibility, dataset_choice, dataset_predefined)
-    dataset_choice.change(upload_visibility, dataset_choice, dataset_upload)
-
-    ##### Model Outputs #####
-
-    # Text output
-    output = gr.Textbox(label="Output")
+def main():
+    with gr.Blocks() as demo:
     
-    ##### Execution #####
-
-    # Setup button
-    tune_btn = gr.Button("Start Fine Tuning")
-    # Execute button
-    tune_btn.click(fn=greet, 
-                   inputs=[model_name, inject_prompt, dataset_predefined],
-                   outputs=output)
+        ##### Title Block #####
+        gr.Markdown("# Instruction Tuning with Unsloth")
+    
+        ##### Model Inputs #####
+    
+        # 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
+        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 Dataset", 
+                                        value='yahma/alpaca-cleaned', 
+                                        visible=True)
+        dataset_upload = gr.UploadButton(label="Upload Dataset (csv, jsonl, or txt)", 
+                                         file_types=[".csv",".jsonl", ".txt"], 
+                                         visible=False)
+        dataset_choice.change(textbox_visibility, 
+                              dataset_choice, 
+                              dataset_predefined)
+        dataset_choice.change(upload_visibility, 
+                              dataset_choice, 
+                              dataset_upload)
+        # Hyperparameters (allow selection, but hide in accordion.)
+        # gr.Accordion('label')
+    
+        
+        ##### Model Outputs #####
+    
+        # Text output
+        output = gr.Textbox(label="Output")
+        
+        ##### Execution #####
+    
+        # Setup button
+        tune_btn = gr.Button("Start Fine Tuning")
+        # Execute button
+        tune_btn.click(fn=greet, 
+                       inputs=[model_name, inject_prompt, dataset_predefined],
+                       outputs=output)
+        # Launch baby
+        demo.launch()
 
 ##################################### Launch #######################################
 
 if __name__ == "__main__":
-    demo.launch()
+    main()