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(
|