Update app.py

app.py

@@ -2,13 +2,13 @@ import gradio as gr
 from transformers import LayoutLMv3Processor, LayoutLMv3ForTokenClassification
 from PIL import Image
 
-# Load the processor and model
+# Load processor and model
 processor = LayoutLMv3Processor.from_pretrained("microsoft/layoutlmv3-base")
 model = LayoutLMv3ForTokenClassification.from_pretrained("microsoft/layoutlmv3-base", num_labels=5)
 
-def extract_invoice_data(image):
-    # Open and preprocess the image
-    image = Image.open(image).convert("RGB")
+def extract_invoice_data(image_path):
+    # Open the image from the file path
+    image = Image.open(image_path).convert("RGB")
     encoding = processor(image, return_tensors="pt")
     outputs = model(**encoding)
     predictions = outputs.logits.argmax(-1).squeeze().tolist()
@@ -17,7 +17,7 @@ def extract_invoice_data(image):
 # Update Gradio interface
 interface = gr.Interface(
     fn=extract_invoice_data,
-    inputs=gr.Image(type="file"),
+    inputs=gr.Image(type="filepath"),  # Use 'filepath' instead of 'file'
     outputs="json",
     title="Invoice Data Extraction"
 )