Update app.py

app.py

@@ -23,18 +23,20 @@ def predict_image(img, conf_threshold, iou_threshold, model_name):
 
     for r in results:
         im_array1 = r.plot(img=img.copy(), labels=False, line_width=4)
-        im1 = Image.fromarray(im_array1[..., ::-1])
+        im1 = Image.fromarray(cv2.cvtColor(im_array1, cv2.COLOR_BGR2RGB))
 
     for r in results:
         im_array2 = r.plot(img=img.copy(), line_width=4, font_size=10)
-        im2 = Image.fromarray(im_array2[..., ::-1])
+        im2 = Image.fromarray(cv2.cvtColor(im_array2, cv2.COLOR_BGR2RGB))
+
+im2 = Image.fromarray(cv2.cvtColor(im_array2, cv2.COLOR_BGR2RGB))
     
     # Calculate the dimensions for the output image
     total_height = im1.height + im2.height
     total_width = im1.width
 
     # Create a blank image with the combined height
-    im = Image.new("BGR", (total_width, total_height))
+    im = Image.new("RGB", (total_width, total_height))
 
     # Paste the images one above the other
     im.paste(im1, (0, 0))