Update app.py

app.py

@@ -30,9 +30,9 @@ def generate_response(prompt, max_tokens, temperature, top_p):
     generated_text = output[0].outputs[0].text
     return generated_text
 
-def ocr_image(image):
-    # Open the image and preprocess for OCR
-    image = Image.open(image)
+def ocr_image(image_path):
+    # Open the image from the file path
+    image = Image.open(image_path)
     pixel_values = ocr_processor(images=image, return_tensors="pt").pixel_values
     
     # Perform OCR
@@ -90,27 +90,27 @@ with gr.Blocks() as demo:
         )
 
     with gr.Tab("OCR"):
-        with gr.Row():
-            with gr.Column():
-                image_input = gr.Image(
-                    label="Upload Image",
-                    type="file",
-                    image_mode="RGB",
-                )
-                ocr_submit_button = gr.Button("Extract Text")
-            
-            with gr.Column():
-                ocr_output = gr.Textbox(
-                    label="Extracted Text",
-                    lines=10,
-                    interactive=False,
-                )
+    with gr.Row():
+        with gr.Column():
+            image_input = gr.Image(
+                label="Upload Image",
+                type="filepath",
+                image_mode="RGB",
+            )
+            ocr_submit_button = gr.Button("Extract Text")
         
-        ocr_submit_button.click(
-            ocr_image,
-            inputs=[image_input],
-            outputs=ocr_output,
-        )
+        with gr.Column():
+            ocr_output = gr.Textbox(
+                label="Extracted Text",
+                lines=10,
+                interactive=False,
+            )
+    
+    ocr_submit_button.click(
+        ocr_image,
+        inputs=[image_input],
+        outputs=ocr_output,
+    )
 
 # Launch the app
 demo.launch()