Update app.py

app.py

@@ -142,7 +142,7 @@ if st.session_state.image is not None:
 
     # predictions
     if 'results' not in st.session_state:
-        st.session_state.results = st.session_state.model.predict(tiles) # predict
+        st.session_state.results = st.session_state.model.predict(tiles, conf=0.5) # predict
     
     
     _, row4_col2, row4_col3, _ = st.columns([1,4,4,1], gap="medium")
@@ -160,7 +160,7 @@ if st.session_state.image is not None:
                     st.session_state.image_counter += 1
                 
         # Plot predicted image
-        im_bgr = st.session_state.results[st.session_state.image_counter].plot(line_width=10)  # BGR-order numpy array
+        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)