w3robotics's picture
Update app.py
4f85dbd verified
raw
history blame
247 Bytes
import streamlit as st
from PIL import Image
# Load model directly
from transformers import AutoTokenizer, AutoModel
tokenizer = AutoTokenizer.from_pretrained("jinhybr/OCR-Donut-CORD")
model = AutoModel.from_pretrained("jinhybr/OCR-Donut-CORD")