Update app.py
Browse files
app.py
CHANGED
@@ -25,7 +25,7 @@ def plot_results(image, results, threshold=0.7):
|
|
25 |
def predict(image):
|
26 |
# make the object detection pipeline
|
27 |
obj_detector = pipeline(
|
28 |
-
"object-detection", model="
|
29 |
)
|
30 |
results = obj_detector(image)
|
31 |
return plot_results(image, results)
|
|
|
25 |
def predict(image):
|
26 |
# make the object detection pipeline
|
27 |
obj_detector = pipeline(
|
28 |
+
"object-detection", model="Antoine101/detr-resnet-50-dc5-hardhat-finetuned"
|
29 |
)
|
30 |
results = obj_detector(image)
|
31 |
return plot_results(image, results)
|