Update train.py

train.py

@@ -129,4 +129,4 @@ with ThreadPoolExecutor(max_workers=num_gpus) as executor:
 ds = datasets.concatenate_datasets(shards)
 
 # %%
-ds.push_to_hub(output_dataset, create_pr=True)
+ds.push_to_hub(output_dataset)