Update app.py

app.py

@@ -1,8 +1,11 @@
 ##################################### Imports ######################################
+# Generic imports
 import gradio as gr
 import json
 import os
 
+# local imports
+
 
 ########################### Global objects and functions ###########################
 
@@ -10,33 +13,51 @@ def get_json_cfg():
     """Retrieve configuration file"""
 
     config_path = os.getenv('CONFIG_PATH')
-    with open('config/config.json', 'r') as file:
+    with open(config_path, 'r') as file:
         config = json.load(file)
 
     return config
 
-print(get_json_cfg())
-dirname = ' '.join(os.listdir())
+
+conf = get_json_cfg()
+
+
+
 
 def greet(model_name, prompt_template, name):
-    return f"Hello {name}!! Using model: {model_name} with template: {prompt_template} and {dirname}"
+    return f"Hello {name}!! Using model: {model_name} with template: {prompt_template}"
 
-model_choices = ["gpt2", "bert-base-uncased", "llama3-8b"]
+model_choices = conf['model']['choices']
 
 
 ##################################### App UI #######################################
 with gr.Blocks() as demo:
+
+    ##### Title Block #####
     gr.Markdown("# Instruction Tuning with Unsloth")
-    model_name = gr.Dropdown(label="Model", choices=model_choices, value="gpt2")
+
+    ##### Model Inputs #####
+
+    # Select Model
+    model_name = gr.Dropdown(label="Model", choices=conf['model']['choices'], value="gpt2")
+    # Show prompt template
     prompt_template = gr.Textbox(label="Prompt Template", value="Instruction: {0}\nOutput: {1}")
+    # Prompt Input
     name_input = gr.Textbox(label="Your Name")
-    
-    greet_btn = gr.Button("Greet")
+
+    ##### Model Outputs #####
+
+    # Text output
     output = gr.Textbox(label="Output")
     
-    greet_btn.click(fn=greet, 
-                    inputs=[model_name, prompt_template, name_input], 
-                    outputs=output)
+    ##### Execution #####
+
+    # Setup button
+    tune_btn = gr.Button("Start Fine Tuning")
+    # Execute button
+    tune_btn.click(fn=greet, 
+                   inputs=[model_name, prompt_template, name_input], 
+                   outputs=output)
 
 
 ##################################### Launch #######################################