Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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=
|
| 26 |
im1 = Image.fromarray(im_array1[..., ::-1])
|
| 27 |
|
| 28 |
for r in results:
|
| 29 |
-
im_array2 = r.plot(img=img.copy(), line_width=
|
| 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
|