Spaces:
Runtime error
Runtime error
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 |
|