ktrapeznikov
/

test-model

Model card Files Files and versions Community

ktrapeznikov commited on Sep 30

Commit

dd3b54d

•

1 Parent(s): bc8a88e

Update script.py

Files changed (1) hide show

script.py +1 -1

script.py CHANGED Viewed

@@ -1,6 +1,6 @@
 import pandas as pd
 from datasets import load_dataset
-dataset_remote = load_dataset("imagefolder",  data_dir="/tmp/data")
 for id in dataset_remote["test"]["id"]:
     out.append(dict(id = id, pred = "flux"))

 import pandas as pd
 from datasets import load_dataset
+dataset_remote = load_dataset("parquet", data_files={'test': '/tmp/data/data/test-00000-of-00001.parquet'})
 for id in dataset_remote["test"]["id"]:
     out.append(dict(id = id, pred = "flux"))