Update app.py

app.py

@@ -44,7 +44,7 @@ if 'image_counter' not in st.session_state:
     st.session_state.image_counter = 0 
 
 if 'confidence' not in st.session_state:
-    st.session_state.confidence = 0.50
+    st.session_state.confidence = 0.70
 
 if 'model' not in st.session_state:
     st.session_state.model = YOLO("/home/user/app/models/head_model/best.pt")  # load 
@@ -217,7 +217,4 @@ if st.session_state.screen == 2:
         # plot predicted image
         im_bgr = st.session_state.results[st.session_state.image_counter].plot()  # BGR-order numpy array
         im_rgb = Image.fromarray(im_bgr[..., ::-1])  # RGB-order PIL image
-        st.image(im_rgb)
-        if 'results' in st.session_state or st.session_state.results != None:
-            st.session_state.confidence = st.slider(label="Change Model Confidence Level and Rerun", min_value=0.01, max_value=0.99, value=st.session_state.confidence, step=0.01)
-            st.session_state.results = None
+        st.image(im_rgb)