Update app.py

app.py

@@ -5,7 +5,9 @@ import json
 
 # Specialized imports
 #from utilities.modeling import modeling
-from datasets import load_dataset
+
+# server import
+from server import get_files, submit_weights #, train_model, submit_weights
 
 
 # Module imports
@@ -39,36 +41,6 @@ class update_visibility:
         else:
             return gr.UploadButton(visible=bool(0))
 
-class get_files:
-
-    def predefined_dataset(dataset_name):
-        global dataset # bad practice, I know... But just bear with me. Will later update to state dict.
-        dataset = load_dataset(dataset_name, split = "train")
-        return 'Successfully loaded dataset'
-    
-    def uploaded_dataset(file):
-        global dataset # bad practice, I know... But just bear with me. Will later update to state dict.
-        dataset = []
-        if file is None:
-            return "File not found. Please upload the file again."
-        try:
-            with open(file,'r') as file:
-                for line in file:
-                    dataset.append(json.loads(line.strip()))
-            return "File retrieved."
-        except FileNotFoundError:
-            return "File not found. Please upload the file again."
-
-    def load_markdown_file(file_path):
-        try:
-            with open(file_path, 'r') as f:
-                return f.read()
-        except FileNotFoundError:
-            return "File not found. Please check the file path."
-        except Exception as e:
-            return f"Error loading file: {str(e)}"
-
-
 def train(model_name, 
           inject_prompt, 
           dataset_predefined,
@@ -93,13 +65,6 @@ def train(model_name,
     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 #######################################