File size: 403 Bytes
064752a
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
#!/bin/bash

base_dir=/path/to/dimt_ocr_free

donut_dir=/path/to/donut-base
nougat_dir=/path/to/nougat-small

checkpoint_dir=/path/to/checkpoint_dir

image_file_path=/path/to/image.png

export CUDA_VISIBLE_DEVICES=0

python inference.py \
    --base_dir $base_dir \
    --donut_dir $donut_dir \
    --nougat_dir $nougat_dir \
    --checkpoint_dir $checkpoint_dir \
    --image_file_path $image_file_path