luisarizmendi
/

hardhat-or-hat

Object Detection

Model card Files Files and versions Community

luisarizmendi commited on Nov 28, 2024

Commit

03cd781

·

verified ·

1 Parent(s): 1f1736a

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -164,7 +164,7 @@ def detect_objects_in_files(files):
         return "No files uploaded.", []
     device = "cuda" if torch.cuda.is_available() else "cpu"
-    model = YOLO("https://github.com/luisarizmendi/ai-apps/raw/refs/heads/main/models/luisarizmendi/object-detector-safety/object-detector-safety-v1.pt")
     model.to(device)
     results_images = []

         return "No files uploaded.", []
     device = "cuda" if torch.cuda.is_available() else "cpu"
+    model = YOLO("https://github.com/luisarizmendi/ai-apps/raw/refs/heads/main/models/luisarizmendi/object-detector-hardhat-or-hat/object-detector-hardhat-or-hat.pt")
     model.to(device)
     results_images = []