Update download_data.py

download_data.py

@@ -0,0 +1,18 @@
+from datasets import load_dataset
+import os
+
+# Load from HF
+dataset = load_dataset("Azu/Handwritten-Mathematical-Expression-Convert-LaTeX")
+
+os.makedirs("dataset/train", exist_ok=True)
+os.makedirs("dataset/val", exist_ok=True)
+
+# Split: 90% train, 10% val
+for i, row in enumerate(dataset["train"]):
+    subset = "val" if i % 10 == 0 else "train"
+    image_path = f"dataset/{subset}/{i:05d}.png"
+    latex_path = f"dataset/{subset}/{i:05d}.txt"
+
+    row["image"].save(image_path)
+    with open(latex_path, "w", encoding="utf-8") as f:
+        f.write(row["label"])