Update app.py

app.py

@@ -16,7 +16,6 @@ from utilities.templates import prompt_template
 
 conf = get_json_cfg()
 
-
 class update_visibility:
 
     def textbox_vis(radio):
@@ -40,15 +39,12 @@ class update_visibility:
         else:
             return gr.UploadButton(visible=bool(0))
 
-
-
 class get_datasets:
 
     def predefined_dataset(dataset_name):
         dataset = load_dataset(dataset_name, split = "train") #dataset_name
         return dataset[0]['output'][0:100]
     
-    
     def uploaded_dataset(file):
         dataset = []
         print(file)
@@ -56,7 +52,7 @@ class get_datasets:
             for line in file:
                 dataset.append(json.loads(line.strip()))
         print(dataset[0]['query'])
-        return dataset[0]['query']
+        return dataset, 'Successfully loaded dataset'
 
 
 
@@ -159,8 +155,6 @@ def main():
 
 
 
-
-
         
 
         ##### Model Parameter Inputs #####