Update model_utils.py

model_utils.py

@@ -3,7 +3,7 @@
 from transformers import ViTForImageClassification
 from PIL import Image
 import torch
-from dataset_utils import load_species_descriptions
+from dataset_utils import DatasetHandler
 
 class BugClassifier:
     def __init__(self, model_path="google/vit-base-patch16-224"):
@@ -15,7 +15,8 @@ class BugClassifier:
             "Luna Moth", "Common Dragonfly", "Honey Bee", "Paper Wasp"
         ]
         # Dynamically load species descriptions
-        self.species_descriptions = load_species_descriptions()
+        handler = DatasetHandler()
+        self.species_descriptions = handler.load_descriptions(max_records=500)
 
     def predict(self, image):
         try: