Update app.py

app.py

@@ -22,8 +22,23 @@ def predict_image(img, conf_threshold, iou_threshold, model_name):
     )
 
     for r in results:
-        im_array = r.plot()
-        im = Image.fromarray(im_array[..., ::-1])
+        im_array1 = r.plot(img=img.copy(), labels=False, line_width=1)
+        im1 = Image.fromarray(im_array1[..., ::-1])
+
+    for r in results:
+        im_array2 = r.plot(img=img.copy(), line_width=1)
+        im2 = Image.fromarray(im_array2[..., ::-1])
+    
+    # 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("RGB", (total_width, total_height))
+
+    # Paste the images one above the other
+    im.paste(im1, (0, 0))
+    im.paste(im2, (0, im1.height))
 
     return im