Maverick98 commited on
Commit
6469b59
·
verified ·
1 Parent(s): a27dab4

update threshold

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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.7):
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