BhumikaMak commited on
Commit
c5224aa
·
1 Parent(s): d4cb7c6

Debug: refactor src

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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(xai_yolov8n(image))
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