Update handler.py

handler.py

@@ -8,8 +8,8 @@ from io import BytesIO
 import numpy as np
 class EndpointHandler():
     def __init__(self, path=""):
-        hf_hub_download(repo_id="Drazcat-AI/flejes", filename="flejes-13/runs/detect/train/weights/best.pt")
-        self.model = YOLO(hf_hub_download(repo_id="Drazcat-AI/flejes", filename="flejes-13/runs/detect/train/weights/best.pt", local_files_only=True))
+        hf_hub_download(repo_id="Drazcat-AI/flejes", filename="flejes-15/runs/detect/train/weights/best.pt")
+        self.model = YOLO(hf_hub_download(repo_id="Drazcat-AI/flejes", filename="flejes-15/runs/detect/train/weights/best.pt", local_files_only=True))
     
     def predict_objects(self, image_path, image_size_m):
         results = self.model(image_path, imgsz=[1280, 960])