Update script.py

script.py

@@ -1,9 +1,8 @@
 import pandas as pd
-test_df = pd.read_csv("/tmp/data/test/metadata.csv")
+from datasets import load_dataset
+dataset_remote = load_dataset("kt-ml/ImageGenTest")
 
-out = []
-
-for file_name in test_df.file_name:
-    out.append(dict(file_name = file_name, pred = "flux"))
+for id in dataset_remote["test"]["id"]:
+    out.append(dict(id = id, pred = "flux"))
 
 pd.Dataframe(out).to_csv("submission.csv",index = False)