Create dataset_utils.py

dataset_utils.py

@@ -0,0 +1,13 @@
+# 2. dataset_utils.py
+# Dataset loading and preprocessing
+from datasets import load_dataset
+from transformers import ViTImageProcessor
+from PIL import Image
+
+def load_insect_dataset():
+    return load_dataset("Francesco/insects-mytwu")
+
+def preprocess_image(image_path):
+    processor = ViTImageProcessor.from_pretrained("google/vit-base-patch16-224")
+    image = Image.open(image_path).convert("RGB")
+    return processor(image, return_tensors="pt")