Update script.py

script.py

@@ -191,7 +191,7 @@ if __name__ == "__main__":
     submission = "submission.csv"
     with open(submission, "w") as f, torch.no_grad():
         writer = csv.writer(f)
-        writer.writerow(["file_name", "text"])
+        writer.writerow(["filename", "text"])
 
         for batch in loader:
             image = batch["image"].float() / 255.0