Update app.py

app.py

@@ -89,7 +89,8 @@ def image_predict(image):
     res = check_acc(box)
     annotated_frame = results[0].plot()
     if len(res) >0:
-        annotated_frame.save("./result/res.png")
+        annotated_frame_bgr = cv2.cvtColor(annotated_frame, cv2.COLOR_RGB2BGR)
+        cv2.imwrite("./result/res.png", annotated_frame_bgr)
         send_email(res)
         return (res, annotated_frame)
 
@@ -119,7 +120,8 @@ def extract_frames(video):
             annotated_frame = results[0].plot()
 
             if len(res) >0:
-                annotated_frame.save("./result/res.png")
+                annotated_frame_bgr = cv2.cvtColor(annotated_frame, cv2.COLOR_RGB2BGR)
+                cv2.imwrite("./result/res.png", annotated_frame_bgr)
                 send_email(res)
                 return (res, annotated_frame)