apailang commited on
Commit
1bc6b45
Β·
1 Parent(s): abfb7d9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -40,9 +40,10 @@ def load_model(model_repo_id):
40
  def predict(pilimg,threshold):
41
 
42
  image_np = pil_image_as_numpy_array(pilimg)
43
- return predict2(image_np,threshold)
 
44
 
45
- def predict2(image_np,threshold):
46
 
47
  results = detection_model(image_np)
48
 
@@ -59,7 +60,7 @@ def predict2(image_np,threshold):
59
  result['detection_scores'][0],
60
  category_index,
61
  use_normalized_coordinates=True,
62
- max_boxes_to_draw=200,
63
  min_score_thresh=threshold,
64
  agnostic_mode=False,
65
  line_thickness=2)
 
40
  def predict(pilimg,threshold):
41
 
42
  image_np = pil_image_as_numpy_array(pilimg)
43
+ # threshold1=threshold
44
+ # return predict2(image_np,threshold)
45
 
46
+ # def predict2(image_np,threshold):
47
 
48
  results = detection_model(image_np)
49
 
 
60
  result['detection_scores'][0],
61
  category_index,
62
  use_normalized_coordinates=True,
63
+ max_boxes_to_draw=20,
64
  min_score_thresh=threshold,
65
  agnostic_mode=False,
66
  line_thickness=2)