Update app.py

app.py

@@ -3,6 +3,8 @@
 import gradio as gr
 import json
 
+# Specialized imports
+#from utilities.modeling import modeling
 
 # Module imports
 from utilities.setup import get_json_cfg
@@ -19,6 +21,7 @@ def textbox_visibility(radio):
     else:
         return gr.Dropdown(visible=bool(0))
 
+
 def upload_visibility(radio):
     value = radio
     if value == "Upload Your Own":
@@ -26,10 +29,49 @@ def upload_visibility(radio):
     else:
         return gr.UploadButton(visible=bool(0))
 
-def greet(model_name, inject_prompt, dataset, pefts):
+def get_dataset(dataset_predefined, dataset_upload, data_field):
+
+    if dataset_predefined.visible:
+        print('')
+    else:
+        print('')
+    return 0
+
+
+
+
+def train(model_name, 
+          inject_prompt, 
+          dataset_predefined,
+          peft,
+          sft,
+          max_seq_length,
+          random_seed,
+          num_epochs,
+          max_steps,
+          data_field,
+          repository,
+          model_out_name):
     """The model call"""
+
+    # Get models
+    # trainer = modeling(model_name, max_seq_length, random_seed,
+    #                    peft, sft, dataset, data_field)
+    # trainer_stats = trainer.train()
+
+    # Return outputs of training.
+    
     return f"Hello!! Using model: {model_name} with template: {inject_prompt}"
 
+
+def submit_weights(model, repository, model_out_name, token):
+    """submits model to repository"""
+    repo = repository + '/' + model_out_name
+    
+    model.push_to_hub(repo, token = token)
+    tokenizer.push_to_hub(repo, token = token)
+    return 0
+
 ##################################### App UI #######################################
 
 def main():
@@ -101,11 +143,18 @@ def main():
         gr.Markdown("### Model Progress")
         # Text output (for now)
         output = gr.Textbox(label="Output") 
+        
+        
+        # Data retrieval
+        
+        
         # Execute button
-        tune_btn.click(fn=greet, 
+        tune_btn.click(fn=train, 
                        inputs=[model_name, 
                                inject_prompt, 
                                dataset_predefined,
+                               dataset_upload,
+                               data_field,
                                peft,
                                sft,
                                max_seq_length,
@@ -117,6 +166,11 @@ def main():
                                model_out_name
                               ],
                        outputs=output)
+        # stop button
+
+        # submit button
+
+        
         # Launch baby
         demo.launch()