Spaces:

opria123
/

community-computer-vision

Sleeping

opria123 commited on Apr 25

Commit

b4a47bb

1 Parent(s): 0255285

Updating requirements

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ model_pipeline = pipeline(
     "object-detection",
     model="opria123/detr-resnet-50-dc5-hardhat-finetuned",
     device=device,
-    threshold=0.1  # Lower threshold to catch more results
 )
 # Function to classify and draw results

     "object-detection",
     model="opria123/detr-resnet-50-dc5-hardhat-finetuned",
     device=device,
+    threshold=0.2  # Lower threshold to catch more results
 )
 # Function to classify and draw results