Drazcat-AI
/

galletas

Model card Files Files and versions

Drazcat-AI commited on Aug 1, 2023

Commit

e6b657c

·

1 Parent(s): 0c24424

Update handler.py

Files changed (1) hide show

handler.py +2 -2

handler.py CHANGED Viewed

@@ -9,8 +9,8 @@ import torch
 import numpy as np
 class EndpointHandler():
     def __init__(self, path=""):
-        hf_hub_download(repo_id="Drazcat-AI/galletas", filename="yolov8_galletas/runs/detect/train/weights/best.pt", local_files_only=True)
-        self.model = YOLO("/home/Drazcat-AI/.cache/huggingface/hub/models--Drazcat-AI--galletas/snapshots/d960a6784ad838e30f01690ce813fb4dc7460bc0/yolov8_galletas/runs/detect/train/weights/best.pt")
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:

 import numpy as np
 class EndpointHandler():
     def __init__(self, path=""):
+        hf_hub_download(repo_id="Drazcat-AI/galletas", filename="yolov8_galletas/runs/detect/train/weights/best.pt")
+        self.model = YOLO(hf_hub_download(repo_id="Drazcat-AI/galletas", filename="yolov8_galletas/runs/detect/train/weights/best.pt", local_files_only=True))
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]: