Trocr Base Printed

🚀 Fork of microsoft/trocr-base-printed for an OCR Inference endpoint

This project is a fork of microsoft/trocr-base-printed, aiming to implement a custom OCR task for 🤗 Inference Endpoints.

🚀 Quick Start

This repository implements a custom task for ocr-detection for 🤗 Inference Endpoints. The code for the customized pipeline is in the pipeline.py.

To use deploy this model as an Inference Endpoint, you have to select Custom as the task to use the pipeline.py file. -> double check if it is selected

💻 Usage Examples

Basic Usage

The endpoint expects the image to be served as binary. Below are cURL and Python examples:

cURL

1. Get the image

wget https://fki.tic.heia-fr.ch/static/img/a01-122-02-00.jpg -O test.jpg

2. Send a cURL request

curl --request POST \
  --url https://{ENDPOINT}/ \
  --header 'Content-Type: image/jpg' \
  --header 'Authorization: Bearer {HF_TOKEN}' \
  --data-binary '@test.jpg'

3. The expected output

{"text": "INDLUS THE"}

Python

1. Get the image

wget https://fki.tic.heia-fr.ch/static/img/a01-122-02-00.jpg -O test.jpg

2. Run the request

import json
from typing import List
import requests as r
import base64

ENDPOINT_URL=""
HF_TOKEN=""

def predict(path_to_image:str=None):
    with open(path_to_image, "rb") as i:
      b = i.read()
    headers= {
        "Authorization": f"Bearer {HF_TOKEN}",
        "Content-Type": "image/jpeg" # content type of image
    }
    response = r.post(ENDPOINT_URL, headers=headers, data=b)
    return response.json()

prediction = predict(path_to_image="test.jpg")

prediction

Expected output

{"text": "INDLUS THE"}