Update utils.py

utils.py

@@ -6,13 +6,12 @@ import uuid
 import pypdfium2 as pdfium
 from PIL import Image
 
-def convert2img(pdf_file):
-    doc = fitz.open(stream=pdf_file.read(), filetype="pdf")
-    page = doc.load_page(0)
-    pix = page.get_pixmap(dpi=150)
-    img = np.frombuffer(pix.samples, dtype=np.uint8).reshape(pix.height, pix.width, pix.n)
-    if img.shape[2] == 4:
-        img = cv2.cvtColor(img, cv2.COLOR_RGBA2BGR)
+def convert2img(path):
+    pdf = pdfium.PdfDocument(path)
+    page = pdf.get_page(0)
+    pil_image = page.render().to_pil()
+    pl1=np.array(pil_image)
+    img = cv2.cvtColor(pl1, cv2.COLOR_RGB2BGR)
     return img
 
 def save_image(img):