Spaces:
Sleeping
Sleeping
Commit
·
c5224aa
1
Parent(s):
d4cb7c6
Debug: refactor src
Browse files
app.py
CHANGED
@@ -15,7 +15,7 @@ def process_image(image, yolo_versions=["yolov5"]):
|
|
15 |
if yolo_version == "yolov5":
|
16 |
result_images.append(xai_yolov5(image))
|
17 |
elif yolo_version == "yolov8s":
|
18 |
-
result_images.append(
|
19 |
else:
|
20 |
result_images.append((Image.fromarray(image), f"{yolo_version} not yet implemented."))
|
21 |
return result_images
|
|
|
15 |
if yolo_version == "yolov5":
|
16 |
result_images.append(xai_yolov5(image))
|
17 |
elif yolo_version == "yolov8s":
|
18 |
+
result_images.append(xai_yolov8s(image))
|
19 |
else:
|
20 |
result_images.append((Image.fromarray(image), f"{yolo_version} not yet implemented."))
|
21 |
return result_images
|