Update app.py

app.py

@@ -78,6 +78,7 @@ async def create_upload_file(text_field: str, file: UploadFile = File(...)):
     if '.json' in file_savePath:
         full_dataset = load_dataset('json', 
                     data_files=file_savePath,
+                    split="train",
                     cache_dir=temp_path,
                     keep_in_memory=True,
                     num_proc=NUM_PROC*2)