easyocr-api / app.py
theoracle's picture
Update app.py
ec903c3 verified
raw
history blame contribute delete
654 Bytes
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)
@spaces.GPU # 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()