Spaces:

shlomihod
/

fm

Runtime error

shlomihod commited on Jun 1, 2024

Commit

10f09b5

1 Parent(s): ccf18cf

trust remote code for datasets

Files changed (1) hide show

app.py CHANGED Viewed

@@ -439,7 +439,7 @@ def prepare_datasets(
         try:
             assert "/" in dataset_name
             dataset_name, subset_name = dataset_name.rsplit("/", 1)
-            ds = load_dataset(dataset_name, subset_name)
         except (FileNotFoundError, AssertionError):
             st.error(f"Dataset `{dataset_name}` not found.")
             st.stop()

         try:
             assert "/" in dataset_name
             dataset_name, subset_name = dataset_name.rsplit("/", 1)
+            ds = load_dataset(dataset_name, subset_name, trust_remote_code=True)
         except (FileNotFoundError, AssertionError):
             st.error(f"Dataset `{dataset_name}` not found.")
             st.stop()