Update app.py

app.py

@@ -24,6 +24,8 @@ from transformers import (
     AutoTokenizer,
 )
 
+from transformers.image_utils import load_image
+
 # --- Constants and Model Setup ---
 MAX_INPUT_TOKEN_LENGTH = 4096
 # Note: The following line correctly falls back to CPU if CUDA is not available.
@@ -236,10 +238,12 @@ def create_gradio_interface():
                              "Nanonets-OCR-s",
                              "Megalodon-OCR-Sync-0713"
                             ],
-                    label="Select Model", value="Nanonets-OCR-s"
+                    label="Select Model", 
+                    value="Nanonets-OCR-s"
                 )
                 task_choice = gr.Dropdown(
-                    choices=["Content Extraction", "Layout Analysis(.json)"],
+                    choices=["Content Extraction", 
+                             "Layout Analysis(.json)"],
                     label="Select Task", value="Content Extraction"
                 )
                 image_input = gr.Image(label="Upload Image", type="pil", sources=['upload'])