Add mask threshold slider

app.py

@@ -259,7 +259,7 @@ def main():
     st.markdown('### Demo')
 
     with st.container():
-        col1, col2, col3 = st.columns(3)
+        col1, col2, col3, col4 = st.columns(4)
 
         with col1:
             option = st.selectbox('Segmentation mode', ('Click', 'Box', 'Everything'))
@@ -269,6 +269,10 @@ def main():
             show_mask = st.checkbox('Show mask',value = True)
 
         with col3:
+            mask_threshold = st.slider('SAM Confidence Threshold',0.0,1.0,0.5,0.05)
+            PREDICTOR_MODEL.model.mask_threshold = mask_threshold
+
+        with col4:
             radius_width = st.slider('Radius/Width for Click/Box',0,20,5,1)
 
     with st.container():