yrodriguezmd commited on
Commit
e799435
·
1 Parent(s): 1735afc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -50,7 +50,8 @@ examples = [["sample_images/"+file] for file in files]
50
  def show_preds(input_image, display_label, display_bbox, detection_threshold):
51
  if detection_threshold==0: detection_threshold=0.5
52
  img = PIL.Image.fromarray(input_image, 'RGB')
53
- pred_dict = model_type.end2end_detect(img, valid_tfms, model, class_map=class_map, detection_threshold=detection_threshold,
 
54
  display_label=display_label, display_bbox=display_bbox, return_img=True,
55
  font_size=16, label_color="#FF59D6")
56
  return pred_dict['img']
 
50
  def show_preds(input_image, display_label, display_bbox, detection_threshold):
51
  if detection_threshold==0: detection_threshold=0.5
52
  img = PIL.Image.fromarray(input_image, 'RGB')
53
+ pred_dict = model_type.end2end_detect(img, valid_tfms, learn, ##model,
54
+ class_map=class_map, detection_threshold=detection_threshold,
55
  display_label=display_label, display_bbox=display_bbox, return_img=True,
56
  font_size=16, label_color="#FF59D6")
57
  return pred_dict['img']