Spaces:

luisarizmendi
/

safety-equipment-object-detection

Sleeping

luisarizmendi commited on Nov 25, 2024

Commit

09076ac

verified ·

1 Parent(s): f0236bb

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,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/safety-hat/safety-hat-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-safety/object-detector-safety-v1.pt")
     model.to(device)
     results_images = []