Update app.py

app.py

@@ -41,16 +41,13 @@ def predict(pilimg,Threshold):
 
     image_np = pil_image_as_numpy_array(pilimg)
     if type(Threshold) is None:
-        Threshold=threshold_d
+        Threshold=threshold_d.astype(int)
         
-    return predict2(image_np,Threshold),predict3(image_np,Threshold),Threshold
+    return predict2(image_np,Threshold),predict3(image_np,Threshold),Threshold.astype(int)
 
 def predict2(image_np,Threshold):
 
     results = detection_model(image_np)
-    
-    if type(Threshold) is None:
-        Threshold=threshold_d
         
     # different object detection models have additional results
     result = {key:value.numpy() for key,value in results.items()}
@@ -66,7 +63,7 @@ def predict2(image_np,Threshold):
         category_index,
         use_normalized_coordinates=True,
         max_boxes_to_draw=20,
-        min_score_thresh=0.0+Threshold,#0.38,
+        min_score_thresh=Threshold,#0.38,
         agnostic_mode=False,
         line_thickness=2)
 
@@ -77,9 +74,6 @@ def predict2(image_np,Threshold):
 
 def predict3(image_np,Threshold):
     
-    if type(Threshold) is None:
-        Threshold=threshold_d
-        
     results = detection_model2(image_np)
 
     # different object detection models have additional results
@@ -96,7 +90,7 @@ def predict3(image_np,Threshold):
         category_index,
         use_normalized_coordinates=True,
         max_boxes_to_draw=20,
-        min_score_thresh=0.0+Threshold,#.38,
+        min_score_thresh=Threshold,#.38,
         agnostic_mode=False,
         line_thickness=2)