fix model not found

app.py

@@ -3,13 +3,8 @@ import khandy
 import numpy as np
 import gradio as gr
 from PIL import Image
-from huggingface_hub import snapshot_download
 from insectid import InsectDetector, InsectIdentifier
-
-MODEL_DIR = snapshot_download(
-    "Genius-Society/insecta",
-    cache_dir="./insectid/__pycache__",
-)
+from utils import MODEL_DIR
 
 
 def infer(filename: str):

insectid/detector.py

@@ -3,15 +3,12 @@ import khandy
 import numpy as np
 from .base import OnnxModel
 from .base import check_image_dtype_and_shape
+from utils import MODEL_DIR
 
 
 class InsectDetector(OnnxModel):
     def __init__(self):
-        current_dir = os.path.dirname(os.path.abspath(__file__))
-        model_path = os.path.join(
-            current_dir,
-            "__pycache__/Genius-Society/insecta/quarrying_insect_detector.onnx",
-        )
+        model_path = f"{MODEL_DIR}/quarrying_insect_detector.onnx"
         self.input_width = 640
         self.input_height = 640
         super(InsectDetector, self).__init__(model_path)

insectid/identifier.py

@@ -5,19 +5,13 @@ import numpy as np
 from .base import OnnxModel
 from collections import OrderedDict
 from .base import check_image_dtype_and_shape
+from utils import MODEL_DIR
 
 
 class InsectIdentifier(OnnxModel):
     def __init__(self):
-        current_dir = os.path.dirname(os.path.abspath(__file__))
-        model_path = os.path.join(
-            current_dir,
-            "__pycache__/Genius-Society/insecta/quarrying_insect_identifier.onnx",
-        )
-        label_map_path = os.path.join(
-            current_dir,
-            "__pycache__/Genius-Society/insecta/quarrying_insectid_label_map.txt",
-        )
+        model_path = f"{MODEL_DIR}/quarrying_insect_identifier.onnx"
+        label_map_path = f"{MODEL_DIR}/quarrying_insectid_label_map.txt"
         super(InsectIdentifier, self).__init__(model_path)
         self.label_name_dict = self._get_label_name_dict(label_map_path)
         self.names = [

utils.py

@@ -0,0 +1,6 @@
+from huggingface_hub import snapshot_download
+
+MODEL_DIR = snapshot_download(
+    "Genius-Society/insecta",
+    cache_dir="./insectid/__pycache__",
+)