Brightsun10 commited on
Commit
c0e65eb
·
verified ·
1 Parent(s): 3115638

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -123,7 +123,7 @@ def predict(input_image):
123
  segment_id = segment["id"]
124
  label_name = model.config.id2label[label_id]
125
 
126
- if score > 0.9 and label_name in TARGET_CLASSES:
127
  # Create a binary mask for this segment
128
  mask = (segmentation == segment_id)
129
 
 
123
  segment_id = segment["id"]
124
  label_name = model.config.id2label[label_id]
125
 
126
+ if score > 0.5 and label_name in TARGET_CLASSES:
127
  # Create a binary mask for this segment
128
  mask = (segmentation == segment_id)
129