Spaces:
Running
on
Zero
Running
on
Zero
import gradio as gr | |
import easyocr | |
import numpy as np | |
from PIL import Image | |
import torch | |
import spaces | |
# Declare reader outside the function (only once) | |
# Ensures this function runs on GPU | |
def extract_text(image): | |
reader = easyocr.Reader(['en'], gpu=True) | |
result = reader.readtext(np.array(image), detail=0) | |
text = "\n".join(result) | |
print("OCR complete. Text:", text) | |
return text | |
iface = gr.Interface( | |
fn=extract_text, | |
inputs=gr.Image(type="pil"), | |
outputs="text", | |
title="GPU OCR with EasyOCR", | |
description="Upload an image. OCR runs on GPU using EasyOCR + Hugging Face Spaces." | |
) | |
iface.launch() | |