| 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")) | |
| pd.DataFrame(out).to_csv("submission.csv",index = False) | |