Update app.py
Browse files
app.py
CHANGED
@@ -9,7 +9,7 @@ description = """
|
|
9 |
|
10 |
Simply upload an image of any person wearning/not-wearing helmet.
|
11 |
"""
|
12 |
-
model_id = "
|
13 |
|
14 |
image_processor = AutoImageProcessor.from_pretrained(model_id)
|
15 |
model = AutoModelForObjectDetection.from_pretrained(model_id)
|
|
|
9 |
|
10 |
Simply upload an image of any person wearning/not-wearing helmet.
|
11 |
"""
|
12 |
+
model_id = "devonho/detr-resnet-50_finetuned_cppe5"
|
13 |
|
14 |
image_processor = AutoImageProcessor.from_pretrained(model_id)
|
15 |
model = AutoModelForObjectDetection.from_pretrained(model_id)
|