Update app.py

app.py

@@ -96,7 +96,11 @@ if input_choice == "📤 Upload Image":
             st.image(result_img, caption="Detection Output")
             if confidence is not None:
                 st.metric("Confidence Score", f"{confidence*100:.2f}%")
-                st.success(label) if "Mask" in label else st.error(label)
+                # st.success(label) if "Mask" in label else st.error(label)
+                if "Mask" in label:
+                    st.success(label)
+                else:
+                    st.error(label)
             else:
                 st.warning(label)
 
@@ -114,9 +118,15 @@ elif input_choice == "📷 Use Webcam":
             result_img, confidence, label = detect_and_predict(image_input)
 
         with col2:
+            st.write("Resulted Image")
             st.image(result_img, caption="Detection Output")
             if confidence is not None:
                 st.metric("Confidence Score", f"{confidence*100:.2f}%")
-                st.success(label) if "Mask" in label else st.error(label)
+                # st.success(label) if "Mask" in label else st.error(label)
+                if "Mask" in label:
+                    st.success(label)
+                else:
+                    st.error(label)
+
             else:
                 st.warning(label)