Spaces:
Runtime error
Runtime error
| import gradio as gr | |
| from gradio_rich_textbox import RichTextbox | |
| from PIL import Image | |
| from surya.ocr import run_ocr | |
| from surya.model.detection.segformer import load_model as load_det_model, load_processor as load_det_processor | |
| from surya.model.recognition.model import load_model as load_rec_model | |
| from surya.model.recognition.processor import load_processor as load_rec_processor | |
| image = Image.open(IMAGE_PATH) | |
| langs = ["en"] # Replace with your languages | |
| det_processor, det_model = load_det_processor(), load_det_model() | |
| rec_model, rec_processor = load_rec_model(), load_rec_processor() | |
| predictions = run_ocr([image], [langs], det_model, det_processor, rec_model, rec_processor) | |