Update model_utils.py

model_utils.py

@@ -4,6 +4,7 @@ from transformers import ViTForImageClassification
 from PIL import Image
 import torch
 from dataset_utils import DatasetHandler
+import threading
 
 class BugClassifier:
     def __init__(self, model_path="google/vit-base-patch16-224"):
@@ -14,9 +15,16 @@ class BugClassifier:
             "Japanese Beetle", "Garden Spider", "Green Grasshopper",
             "Luna Moth", "Common Dragonfly", "Honey Bee", "Paper Wasp"
         ]
-        # Dynamically load species descriptions
-        handler = DatasetHandler()
-        self.species_descriptions = handler.load_descriptions(max_records=500)
+        self.species_descriptions = {}
+        self.load_species_descriptions()
+
+    def load_species_descriptions(self):
+        def load():
+            handler = DatasetHandler()
+            self.species_descriptions = handler.load_descriptions(max_records=500)
+
+        thread = threading.Thread(target=load)
+        thread.start()
 
     def predict(self, image):
         try:
@@ -39,4 +47,4 @@ class BugClassifier:
     def get_species_info(self, species):
         return self.species_descriptions.get(
             species, "Information not available. Consider updating your dataset for this species."
-        )
+        )