Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -54,7 +54,7 @@ def show_preds(input_image):
|
|
| 54 |
img = PIL.Image.fromarray(input_image, "RGB")
|
| 55 |
|
| 56 |
pred_dict = model_type.end2end_detect(img, valid_tfms, model, class_map=class_map, detection_threshold=0.5,
|
| 57 |
-
display_label=
|
| 58 |
font_size=16, label_color="#FF59D6")
|
| 59 |
|
| 60 |
#pred_dict = model_type.end2end_detect(
|
|
|
|
| 54 |
img = PIL.Image.fromarray(input_image, "RGB")
|
| 55 |
|
| 56 |
pred_dict = model_type.end2end_detect(img, valid_tfms, model, class_map=class_map, detection_threshold=0.5,
|
| 57 |
+
display_label=False, display_bbox=True, return_img=True,
|
| 58 |
font_size=16, label_color="#FF59D6")
|
| 59 |
|
| 60 |
#pred_dict = model_type.end2end_detect(
|