Skip to content

Instantly share code, notes, and snippets.

@mht-sharma
mht-sharma / onnx_trocr_inference.py
Created December 16, 2022 10:46
ONNX TrOCR Inference
import os
import time
from typing import Optional, Tuple
import torch
from PIL import Image
import onnxruntime as onnxrt
import requests
from transformers import AutoConfig, AutoModelForVision2Seq, TrOCRProcessor, VisionEncoderDecoderModel