Update app.py

app.py

@@ -51,6 +51,9 @@ def predict(pilimg,Threshold):
 def predict2(image_np,Threshold):
 
     results = detection_model(image_np)
+    
+    if Threshold is None or Threshold == 0:
+        Threshold=threshold_d
         
     # different object detection models have additional results
     result = {key:value.numpy() for key,value in results.items()}
@@ -66,7 +69,7 @@ def predict2(image_np,Threshold):
         category_index,
         use_normalized_coordinates=True,
         max_boxes_to_draw=20,
-        min_score_thresh=threshold_d,#0.38,
+        min_score_thresh=threshold,#0.38,
         agnostic_mode=False,
         line_thickness=2)
 
@@ -93,7 +96,7 @@ def predict3(image_np,Threshold):
         category_index,
         use_normalized_coordinates=True,
         max_boxes_to_draw=20,
-        min_score_thresh=threshold_d,#.38,
+        min_score_thresh=threshold,#.38,
         agnostic_mode=False,
         line_thickness=2)