BhumikaMak commited on
Commit
abce576
·
1 Parent(s): 279313f

Debug: yolov8 version incompatibility

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ def process_image(image, yolo_versions=["yolov5"]):
14
  for yolo_version in yolo_versions:
15
  if yolo_version == "yolov5":
16
  result_images.append(xai_yolov5(image))
17
- elif yolo_version == "yolov8":
18
  result_images.append(xai_yolov8n(image))
19
  else:
20
  result_images.append((Image.fromarray(image), f"{yolo_version} not yet implemented."))
 
14
  for yolo_version in yolo_versions:
15
  if yolo_version == "yolov5":
16
  result_images.append(xai_yolov5(image))
17
+ elif yolo_version == "yolov8n":
18
  result_images.append(xai_yolov8n(image))
19
  else:
20
  result_images.append((Image.fromarray(image), f"{yolo_version} not yet implemented."))