Update app.py

app.py

@@ -3,6 +3,7 @@ from transformers import AutoProcessor, AutoModel
 from PIL import Image
 from pdf2image import convert_from_path
 from PyPDF2 import PdfReader
+import os
 
 # Load the LayoutLMv3 model and processor
 processor = AutoProcessor.from_pretrained("microsoft/layoutlmv3-base")
@@ -31,9 +32,10 @@ def process_image(image):
 
 # Gradio interface function
 def extract_table(file, page_number):
-    if file.type == "application/pdf":
+    file_ext = os.path.splitext(file.name)[1].lower()
+    if file_ext == ".pdf":
         return process_pdf(file, page_number)
-    elif file.type.startswith("image"):
+    elif file_ext in [".png", ".jpg", ".jpeg"]:
         image = Image.open(file)
         return process_image(image)
     else: