Update app.py

app.py

@@ -280,7 +280,8 @@ def plot_image(image, boxes, image_ori=None):
 def infer(model, img, thresh, iou_thresh, anchors):
     model.eval()
     image = np.array(img)
-    print(image.shape)
+    image = image[:, :, :3]
+    # print(image.shape)
     image_copy = image.copy()
     # image = image[np.newaxis, :]
     augmentations = infer_transforms(image=image)
@@ -350,7 +351,7 @@ scaled_anchors = (
 
 uploaded_file = st.file_uploader("Chọn hình ảnh...", type=["jpg", "jpeg", "png"])
 # uploaded_file = '/home/ngocanh/Documents/final_thesis/code/dataset/10_10/base/images/test/000011.jpg'
-if uploaded_file is not None:ssssssssssssssssssssssssssss
+if uploaded_file is not None:
     image = Image.open(uploaded_file)
     print("Thuc hien bien doi")