Update app.py

app.py

@@ -108,7 +108,7 @@ def process_image():
             image_np = np.array(image)
             
             # Hiển thị ảnh gốc
-            st.image(image_np, caption="Ảnh đã tải lên", use_column_width=True)
+            st.image(image_np, caption="Ảnh đã tải lên", use_container_width=True)
     else:
         # Sử dụng ảnh mẫu
         sample_img_url = "https://storage.googleapis.com/sfr-vision-language-research/DINO/ground_truth_images/000000014439.jpg"
@@ -118,7 +118,7 @@ def process_image():
             image_np = np.array(image)
             
             # Hiển thị ảnh mẫu
-            st.image(image_np, caption="Ảnh mẫu", use_column_width=True)
+            st.image(image_np, caption="Ảnh mẫu", use_container_width=True)
         except Exception as e:
             st.error(f"Không thể tải ảnh mẫu: {e}")
     
@@ -210,8 +210,8 @@ def process_image():
             st.success(f"Đã phát hiện {person_count} người trong ảnh")
             
             col1, col2 = st.columns(2)
-            col1.image(output_image, caption="Phát hiện người với độ sâu", use_column_width=True)
-            col2.image(depth_colormap, caption="Bản đồ độ sâu", use_column_width=True)
+            col1.image(output_image, caption="Phát hiện người với độ sâu", use_container_width=True)
+            col2.image(depth_colormap, caption="Bản đồ độ sâu", use_container_width=True)
 
 # Chức năng xử lý video
 def process_video():
@@ -381,8 +381,8 @@ def process_video():
                 
                 # Hiển thị khung hình trong Streamlit (cập nhật theo khoảng)
                 if frame_counter % 5 == 0 or frame_counter == 1:
-                    detection_placeholder.image(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB), caption="Phát hiện người", use_column_width=True)
-                    depth_placeholder.image(depth_colormap, caption="Bản đồ độ sâu", use_column_width=True)
+                    detection_placeholder.image(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB), caption="Phát hiện người", use_container_width=True)
+                    depth_placeholder.image(depth_colormap, caption="Bản đồ độ sâu", use_container_width=True)
                 
                 # Ghi khung hình vào videos
                 out_detection.write(frame)