Spaces:
Running
Running
Update app.py
Browse files
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 |
-
|
44 |
-
if type(
|
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 |
|