Update app.py

app.py

@@ -18,7 +18,7 @@ from ultralytics import YOLO
 from tensorflow.keras.applications.vgg16 import VGG16, preprocess_input as vgg16_preprocess
 from tensorflow.keras.applications.efficientnet import EfficientNetB0, preprocess_input as effnet_preprocess
 
-# ------------------------------------------------------------
+# ------------------------------------------------------------use
 # GLOBAL CONFIG
 # ------------------------------------------------------------
 st.set_page_config(
@@ -709,10 +709,10 @@ YOLOv8 will detect all objects and optionally verify them with the best classifi
         col1, col2 = st.columns(2)
 
         with col1:
-            st.image(pil_img, caption="Uploaded Image", width='stretch')
+            st.image(pil_img, caption="Uploaded Image", width='content')
 
         with col2:
-            st.image(result["annotated_image"], caption="Detected Result", width='stretch')
+            st.image(result["annotated_image"], caption="Detected Result", width='content')
 
         st.write(f"YOLO inference time: {result['yolo_inference_time_sec']*1000:.1f} ms")
         st.write(f"Number of detections: {len(result['detections'])}")
@@ -749,18 +749,18 @@ elif page == "📊 Model Performance":
         with col1:
             st.bar_chart(
                 df_cls.set_index("Model")["Accuracy"],
-                width='stretch',
+                width='content',
             )
         with col2:
             st.bar_chart(
                 df_cls.set_index("Model")["F1 (weighted)"],
-                width='stretch',
+                width='content',
             )
 
         st.markdown("#### Inference Speed (images/sec)")
         st.bar_chart(
             df_cls.set_index("Model")["Images/sec"],
-            width='stretch',
+            width='content',
         )
 
     # --- YOLO metrics ---
@@ -791,7 +791,7 @@ elif page == "📊 Model Performance":
         ]
         if imgs:
             for img in sorted(imgs):
-                st.image(img, caption=os.path.basename(img), width='stretch')
+                st.image(img, caption=os.path.basename(img), width='content')
         else:
             st.info("No comparison plots found in `smartvision_metrics/comparison_plots/`.")
     else: