Update app.py

app.py

@@ -868,8 +868,7 @@ def main():
             image = uploaded_image
             col1, col2 = st.columns([1, 2])
             with col1:
-                # Use a consistent width based on the container
-                st.image(image, caption="Uploaded Image", use_column_width=True)
+                st.image(image, caption="Uploaded Image", width=300)
             
             # Continue with Xception model analysis
             if st.session_state.xception_model_loaded:
@@ -916,23 +915,6 @@ def main():
                         with col2:
                             st.markdown("### Detection Result")
                             st.markdown(f"**Classification:** {pred_label} (Confidence: {confidence:.2%})")
-                            
-                            # Display face box on image if detected
-                            if face_box:
-                                # Create a proper visualization that matches the GradCAM alignment
-                                img_to_show = original_image.copy()
-                                img_draw = np.array(img_to_show)
-                                x, y, w, h = face_box
-                                
-                                # Draw rectangle with thicker line and better color for visibility
-                                cv2.rectangle(img_draw, (x, y), (x + w, y + h), (0, 255, 0), 3)
-                                
-                                # Add a caption inside the image for clarity
-                                font = cv2.FONT_HERSHEY_SIMPLEX
-                                cv2.putText(img_draw, "Detected Face", (x, y-10), font, 0.6, (0, 255, 0), 2)
-                                
-                                # Display full image with box instead of cropped face
-                                st.image(Image.fromarray(img_draw), caption="Face Detection Result", use_column_width=True)
                         
                         # GradCAM visualization with error handling
                         st.subheader("GradCAM Visualization")
@@ -942,11 +924,8 @@ def main():
                             )
                             
                             if comparison:
-                                # Add some spacing for better visual separation
-                                st.write("")
-                                
-                                # Display GradCAM results with better sizing and alignment
-                                st.image(comparison, caption="Original | CAM | Overlay", use_column_width=True)
+                                # Display GradCAM results (controlled size)
+                                st.image(comparison, caption="Original | CAM | Overlay", width=700)
                                 
                                 # Save for later use
                                 st.session_state.comparison_image = comparison