Update app.py

app.py

@@ -42,8 +42,9 @@ class update_visibility:
 class get_datasets:
 
     def predefined_dataset(dataset_name):
-        dataset = load_dataset(dataset_name, split = "train") #dataset_name
-        return dataset, 'Successfully loaded dataset'
+        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.
@@ -54,8 +55,7 @@ class get_datasets:
             with open(file,'r') as file:
                 for line in file:
                     dataset.append(json.loads(line.strip()))
-            print(dataset[0]['query'])
-            return "File retrieved"
+            return "File retrieved."
         except FileNotFoundError:
             return "File not found. Please upload the file again."
 
@@ -122,10 +122,6 @@ def main():
         dataset_predefined = gr.Textbox(label="Hugging Face Hub Training Dataset", 
                                         value='yahma/alpaca-cleaned', 
                                         visible=True)
-
-        #dataset_uploaded_load = gr.File(label="Upload Dataset", 
-        #                                file_types=[".csv",".jsonl", ".txt"], 
-        #                                visible=False)
         
         dataset_predefined_load = gr.Button("Upload Dataset (.csv, .jsonl, or .txt)")
 
@@ -133,7 +129,6 @@ def main():
                                          file_types=[".csv",".jsonl", ".txt"], 
                                          visible=False)
         data_snippet = gr.Markdown()
-
         
         dataset_choice.change(update_visibility.textbox_vis, 
                               dataset_choice,