fix app py

.gitignore

@@ -3,3 +3,4 @@ _local/
 *.pyc
 local_models_*/
 rename.sh
+*.onnx

app.py

@@ -1,12 +1,43 @@
+import os
 import cv2
 import khandy
+import requests
 import numpy as np
 import gradio as gr
+from tqdm import tqdm
 from PIL import Image
 from insectid import InsectDetector
 from insectid import InsectIdentifier
 
 
+def download_model(url, local_path):
+    # Check if the file exists
+    if not os.path.exists(local_path):
+        print(f"Downloading file from {url}...")
+        # Make a request to the URL
+        response = requests.get(url, stream=True)
+
+        # Get the total file size in bytes
+        total_size = int(response.headers.get('content-length', 0))
+
+        # Initialize the tqdm progress bar
+        progress_bar = tqdm(total=total_size, unit='B', unit_scale=True)
+
+        # Open a local file with write-binary mode
+        with open(local_path, 'wb') as file:
+            for data in response.iter_content(chunk_size=1024):
+                # Update the progress bar
+                progress_bar.update(len(data))
+
+                # Write the data to the local file
+                file.write(data)
+
+        # Close the progress bar
+        progress_bar.close()
+
+        print("Download completed.")
+
+
 def inference(filename):
     detector = InsectDetector()
     identifier = InsectIdentifier()
@@ -66,6 +97,16 @@ def inference(filename):
     return Image.fromarray(image_for_draw[:, :, ::-1], mode='RGB')
 
 
+domain = "https://huggingface.co/MuGeminorum/insecta/resolve/main/"
+download_model(
+    f"{domain}quarrying_insect_detector.onnx",
+    "./insectid/models/quarrying_insect_detector.onnx"
+)
+download_model(
+    f"{domain}quarrying_insect_identifier.onnx",
+    "./insectid/models/quarrying_insect_identifier.onnx"
+)
+
 with gr.Blocks() as demo:
     with gr.Tab("Image"):
         gr.Markdown("## Insect Inference on Image")