encode

app/app.py

@@ -64,8 +64,9 @@ def doc_layout():
                 if r["type"] == "plain text":
                     x, y, x1, y1 = r["coords"]
                     x, y, x1, y1 = int(x), int(y), int(x1), int(y1)
-                    pbytes = pimg[y:y1,x:x1].tobytes()
-                    single_img_doc = DocumentFile.from_images(pbytes)
+                    block = pimg[y:y1,x:x1]
+                    img_encoded = cv2.imencode('.png', block)[1].tobytes()
+                    single_img_doc = DocumentFile.from_images(img_encoded)
                     result = doctr_model(single_img_doc)
                     print(result)