Update app.py

app.py

@@ -79,16 +79,18 @@ input_choice = st.selectbox("Choose Input Method", ["📤 Upload Image", "📷 U
 if input_choice == "📤 Upload Image":
     uploaded_file = st.file_uploader("Choose an image file", type=["jpg", "jpeg", "png"])
     if uploaded_file:
+        col1, col2 = st.columns(2)
         image_input = resize_image(Image.open(uploaded_file))
-        st.image(image_input, caption="Uploaded Image", use_container_width=True)
+        with col1:
+            st.image(image_input, caption="Uploaded Image", width=400,height = 150)#use_container_width=True)
 
         with st.spinner("Analyzing with AI model..."):
             result_img, confidence, label = detect_and_predict(image_input)
 
-        col1, col2 = st.columns(2)
-        with col1:
-            st.image(result_img, caption="Detection Output", use_container_width=True)
+        # col1, col2 = st.columns(2)
         with col2:
+            st.image(result_img, caption="Detection Output", width=400,height = 150)#use_container_width=True)
+        # with col2:
             if confidence is not None:
                 st.metric("Confidence Score", f"{confidence*100:.2f}%")
                 if "Mask" in label:
@@ -113,9 +115,9 @@ elif input_choice == "📷 Use Webcam":
 
         # col1, col2 = st.columns(2)
         with col2:
-            resized_img = result_img.resize((200, 200))
+            # resized_img = result_img.resize((200, 200))
 
-            st.image(resized_img, caption="Detection Output", width=200)
+            st.image(resized_img, caption="Detection Output", width=400,height = 150)
             # st.image(result_img, caption="Detection Output", use_container_width=True).resize(200,200)
         # with col2:
             if confidence is not None: