Update app.py

app.py

@@ -58,11 +58,15 @@ def show_preds(input_image, display_label, display_bbox, detection_threshold):
     if detection_threshold==0: detection_threshold=0.5
 
     img = PIL.Image.fromarray(input_image, 'RGB')
-
-    pred_dict  = model_type.end2end_detect(img, valid_tfms, model, class_map=class_map, detection_threshold=detection_threshold,
-                                           display_label=display_label, display_bbox=display_bbox, return_img=True, 
+    
+    pred_dict  = model_type.end2end_detect(img, valid_tfms, model, class_map=class_map, detection_threshold=0.5,
+                                           display_label=True, display_bbox=True, return_img=True, 
                                            font_size=16, label_color="#FF59D6")
 
+    #pred_dict  = model_type.end2end_detect(img, valid_tfms, model, class_map=class_map, detection_threshold=detection_threshold,
+    #                                       display_label=display_label, display_bbox=display_bbox, return_img=True, 
+    #                                       font_size=16, label_color="#FF59D6")
+
     return pred_dict['img']
 
 # display_chkbox = gr.inputs.CheckboxGroup(["Label", "BBox"], label="Display", default=True)