ajcasagrande commited on
Commit
d39d160
·
verified ·
1 Parent(s): 98aca02

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -22,11 +22,11 @@ def predict_image(img, conf_threshold, iou_threshold, model_name):
22
  )
23
 
24
  for r in results:
25
- im_array1 = r.plot(img=img.copy(), labels=False, line_width=1)
26
  im1 = Image.fromarray(im_array1[..., ::-1])
27
 
28
  for r in results:
29
- im_array2 = r.plot(img=img.copy(), line_width=1)
30
  im2 = Image.fromarray(im_array2[..., ::-1])
31
 
32
  # Calculate the dimensions for the output image
 
22
  )
23
 
24
  for r in results:
25
+ im_array1 = r.plot(img=img.copy(), labels=False, line_width=2)
26
  im1 = Image.fromarray(im_array1[..., ::-1])
27
 
28
  for r in results:
29
+ im_array2 = r.plot(img=img.copy(), line_width=2)
30
  im2 = Image.fromarray(im_array2[..., ::-1])
31
 
32
  # Calculate the dimensions for the output image