Spaces:
Runtime error
Runtime error
polejowska
commited on
Commit
•
a1b8785
1
Parent(s):
340429d
Update app.py
Browse files
app.py
CHANGED
@@ -41,6 +41,7 @@ def detect_objects(model_name, image_input, threshold):
|
|
41 |
model_details = "DETR details"
|
42 |
elif "YOLOS" in model_name:
|
43 |
model = YolosForObjectDetection.from_pretrained(MODELS_REPO[model_name])
|
|
|
44 |
|
45 |
(
|
46 |
processed_outputs,
|
|
|
41 |
model_details = "DETR details"
|
42 |
elif "YOLOS" in model_name:
|
43 |
model = YolosForObjectDetection.from_pretrained(MODELS_REPO[model_name])
|
44 |
+
model_details = "YOLOS details"
|
45 |
|
46 |
(
|
47 |
processed_outputs,
|