erax commited on
Commit
f3ee0fd
·
verified ·
1 Parent(s): bcec406

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -73,7 +73,7 @@ def yolov8_inference(
73
  # results = model.predict(image, imgsz=image_size)
74
  # render = render_result(model=model, image=image, result=results[0])
75
 
76
- return Image.fromarray(annotated_image, 'RGB')
77
 
78
 
79
  inputs = [
 
73
  # results = model.predict(image, imgsz=image_size)
74
  # render = render_result(model=model, image=image, result=results[0])
75
 
76
+ return annotated_image[:, :, ::-1]
77
 
78
 
79
  inputs = [