apailang commited on
Commit
7032a5d
β€’
1 Parent(s): 83beac5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -40,9 +40,11 @@ def load_model(model_repo_id):
40
  def predict(pilimg,Threshold):
41
 
42
  image_np = pil_image_as_numpy_array(pilimg)
43
- Threshold1= float(Threshold)
44
- if type(Threshold1) is None or Threshold1 == 0:
45
  Threshold=0.88
 
 
46
 
47
  return predict2(image_np,Threshold),predict3(image_np,Threshold),Threshold
48
 
 
40
  def predict(pilimg,Threshold):
41
 
42
  image_np = pil_image_as_numpy_array(pilimg)
43
+
44
+ if type(Threshold) is None or Threshold == 0:
45
  Threshold=0.88
46
+ else:
47
+ Threshold= float(Threshold)
48
 
49
  return predict2(image_np,Threshold),predict3(image_np,Threshold),Threshold
50