Update app.py

app.py

@@ -1,7 +1,7 @@
-import gradio as gr
+import torch
 from transformers import AutoModel, AutoTokenizer
 from PIL import Image
-import torch
+import gradio as gr
 
 # Load the OCR model and tokenizer
 tokenizer = AutoTokenizer.from_pretrained('ucaslcl/GOT-OCR2_0', trust_remote_code=True)
@@ -19,14 +19,14 @@ def perform_ocr(image):
     result = model.chat(tokenizer, image, ocr_type='ocr')
     return result
 
-# Create the Gradio interface
+# Create the Gradio interface using the new syntax
 interface = gr.Interface(
     fn=perform_ocr, 
-    inputs=gr.inputs.Image(type="pil"),  # Accepts an image input
-    outputs="text",  # Outputs extracted text
+    inputs=gr.Image(type="pil"),  # Updated to gr.Image
+    outputs=gr.Textbox(),         # Updated to gr.Textbox
     title="OCR Web App",
     description="Upload an image to extract text using the GOT-OCR2.0 model."
 )
 
 # Launch the app
-interface.launch()
+interface.launch(share=True)