app.py

# Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("stepfun-ai/GOT-OCR2_0", trust_remote_code=True)