Update app.py

app.py

@@ -364,6 +364,7 @@ if uploaded_file is not None:
     image_1 = infer(model, image, 0.5, 0.5, scaled_anchors)
     
     #task 2
+    image = Image.open(uploaded_file)
     file_path = f"2007_task2_{base}_{new}_mAP_{base}_{new}.pth.tar"
     model = YOLOv3(num_classes=all).to(device)
     checkpoint = torch.load(file_path, map_location=device)
@@ -372,6 +373,7 @@ if uploaded_file is not None:
     image_2 = infer(model, image, 0.5, 0.5, scaled_anchors)
     
     #ft
+    image = Image.open(uploaded_file)
     file_path = f"2007_finetune_{base}_{new}_mAP_{base}_{new}.pth.tar"
     checkpoint = torch.load(file_path, map_location=device)
     model.load_state_dict(checkpoint["state_dict"])
@@ -383,7 +385,9 @@ if uploaded_file is not None:
     # note = Image.open("note.png")
     # st.image(note, width=150)
     
-    
+    image_1 = cv2.cvtColor(image_1, cv2.COLOR_BGR2RGB)
+    image_2 = cv2.cvtColor(image_2, cv2.COLOR_BGR2RGB)
+    image_3 = cv2.cvtColor(image_3, cv2.COLOR_BGR2RGB)
     col1, col2, col3, col4 = st.columns(4)
     with col1:
         st.image(image, caption="Ảnh đầu vào", use_column_width=True)