Spaces:
Running
Running
update threshold
Browse files
app.py
CHANGED
@@ -85,7 +85,7 @@ def load_image(image_path_or_url):
|
|
85 |
image = image.unsqueeze(0) # Add batch dimension
|
86 |
return image
|
87 |
|
88 |
-
def predict(image_path_or_file, title, threshold=0.
|
89 |
# Preprocess the image
|
90 |
image = load_image(image_path_or_file)
|
91 |
|
|
|
85 |
image = image.unsqueeze(0) # Add batch dimension
|
86 |
return image
|
87 |
|
88 |
+
def predict(image_path_or_file, title, threshold=0.4):
|
89 |
# Preprocess the image
|
90 |
image = load_image(image_path_or_file)
|
91 |
|