File size: 677 Bytes
c7d972f 0db2578 c06a6ff 0db2578 c06a6ff 0db2578 198791b f8c03e7 198791b 0db2578 198791b f8c03e7 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 | import gradio as gr
from transformers import pipeline
# 1. Carregar o Pipeline de OCR
# O nome real do modelo no HF Hub pode ser diferente, pesquise por 'got-ocr-20'
pipe = pipeline("image-to-text", model="NOME_REAL_DO_MODELO_HF")
def extract_text_local(image):
# 2. Executar o modelo na imagem
result = pipe(image)
return result[0]['generated_text'] if result else "Nenhum texto encontrado."
# 3. Criar a Interface Gradio
iface = gr.Interface(
fn=extract_text_local,
inputs=gr.Image(type="pil", label="Carregue uma imagem com texto"), # 'pil' para objeto Pillow
outputs="text",
title="OCR com got-ocr-20 (Implantação Local)"
)
iface.launch() |