Update app.py

app.py

@@ -43,16 +43,19 @@ class get_datasets:
 
     def predefined_dataset(dataset_name):
         dataset = load_dataset(dataset_name, split = "train") #dataset_name
-        return dataset[0]['output'][0:100]
+        return dataset, 'Successfully loaded dataset'
     
     def uploaded_dataset(file):
         dataset = []
         print(file)
-        with open(file,'r') as file:
-            for line in file:
-                dataset.append(json.loads(line.strip()))
-        print(dataset[0]['query'])
-        return dataset, 'Successfully loaded dataset'
+        try:
+            with open(file,'r') as file:
+                for line in file:
+                    dataset.append(json.loads(line.strip()))
+            print(dataset[0]['query'])
+            return dataset
+        except FileNotFoundError:
+            return "File not found. Please upload the file again."
 
 
 
@@ -122,14 +125,13 @@ def main():
         #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)")
 
         dataset_uploaded_load = gr.UploadButton(label="Upload Dataset (.csv, .jsonl, or .txt)", 
                                          file_types=[".csv",".jsonl", ".txt"], 
                                          visible=False)
-      
+
         data_field = gr.Textbox(label="Dataset Training Field",
                                 value=conf['model']['general']["dataset_text_field"])