Spaces:
Sleeping
Sleeping
Updating requirements
Browse files
app.py
CHANGED
|
@@ -12,7 +12,7 @@ model_pipeline = pipeline(
|
|
| 12 |
"object-detection",
|
| 13 |
model="opria123/detr-resnet-50-dc5-hardhat-finetuned",
|
| 14 |
device=device,
|
| 15 |
-
threshold=0.
|
| 16 |
)
|
| 17 |
|
| 18 |
# Function to classify and draw results
|
|
|
|
| 12 |
"object-detection",
|
| 13 |
model="opria123/detr-resnet-50-dc5-hardhat-finetuned",
|
| 14 |
device=device,
|
| 15 |
+
threshold=0.2 # Lower threshold to catch more results
|
| 16 |
)
|
| 17 |
|
| 18 |
# Function to classify and draw results
|