apailang commited on
Commit
312651e
β€’
1 Parent(s): 480ce71

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -40,7 +40,7 @@ def load_model(model_repo_id):
40
  def predict(pilimg,Threshold):
41
 
42
  image_np = pil_image_as_numpy_array(pilimg)
43
- if type(Threshold) is None:
44
  Threshold=threshold_d
45
 
46
  return predict2(image_np,Threshold),predict3(image_np,Threshold),Threshold
 
40
  def predict(pilimg,Threshold):
41
 
42
  image_np = pil_image_as_numpy_array(pilimg)
43
+ if type(Threshold) is None or Threshold==0:
44
  Threshold=threshold_d
45
 
46
  return predict2(image_np,Threshold),predict3(image_np,Threshold),Threshold