Spaces:

luisarizmendi
/

object-detection-batch

Sleeping

luisarizmendi commited on Nov 29, 2024

Commit

82609ae

verified ·

1 Parent(s): b5c9abe

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ import os
 import cv2
 import torch
-DEFAULT_MODEL_URL = "https://github.com/luisarizmendi/ai-apps/raw/refs/heads/main/models/luisarizmendi/object-detector-hardhat-or-hat/object-detector-hardhat-or-hat.pt"
 def detect_objects_in_files(model_input, files):
     """

 import cv2
 import torch
+DEFAULT_MODEL_URL = "https://github.com/luisarizmendi/ai-apps/raw/refs/heads/main/models/luisarizmendi/object-detection-hardhat-or-hat/object-detection-hardhat-or-hat-m.pt"
 def detect_objects_in_files(model_input, files):
     """