Update app.py

app.py

@@ -64,16 +64,32 @@ def main():
         dataset_choice.change(upload_visibility, 
                               dataset_choice, 
                               dataset_upload)
+        # Parameters
+        max_seq_length = gr.Textbox(label="Maximum sequence length", 
+                                     value=conf['model']['general']["max_seq_length"])
+        random_seed = gr.Textbox(label="Seed",
+                                value=conf['model']['general']["seed"])
+        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"])   
+        data_field = gr.Textbox(label="Dataset Training Field",
+                                value=conf['model']['general']["dataset_text_field"])
+        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"])   
+
         # Hyperparameters (allow selection, but hide in accordion.)
         with gr.Accordion("Advanced Tuning", open=False):
 
-            sftparams = conf['model']['sft']
+            sftparams = conf['model']['general']
             # accordion container content
             dict_string = json.dumps(dict(conf['model']['peft']), indent=4)
             peft = gr.Textbox(label="PEFT Parameters (json)", value=dict_string)
             
             dict_string = json.dumps(dict(conf['model']['sft']), indent=4)
-            peft = gr.Textbox(label="SFT Parameters (json)", value=dict_string)            
+            sft = gr.Textbox(label="SFT Parameters (json)", value=dict_string)            
         
         ##### Execution #####
     
@@ -86,8 +102,15 @@ def main():
                        inputs=[model_name, 
                                inject_prompt, 
                                dataset_predefined,
-                               peft
-                               # sft
+                               peft,
+                               sft,
+                               max_seq_length,
+                               random_seed,
+                               num_epochs,
+                               max_steps,
+                               data_field,
+                               repository,
+                               model_out_name
                               ],
                        outputs=output)
         # Launch baby