likhithv commited on
Commit
c6e3997
1 Parent(s): feaa33a

Update donut_inference.py

Browse files
Files changed (1) hide show
  1. donut_inference.py +2 -2
donut_inference.py CHANGED
@@ -11,10 +11,10 @@ load_dotenv()
11
 
12
  device = "cuda" if torch.cuda.is_available() else "cpu"
13
  # Load the processor from the local directory
14
- processor = DonutProcessor.from_pretrained("/DocQA/Model")
15
 
16
  # Load the model from the local directory
17
- model = VisionEncoderDecoderModel.from_pretrained("/DocQA/Model")
18
  model.to(device)
19
 
20
  @st.cache_resource
 
11
 
12
  device = "cuda" if torch.cuda.is_available() else "cpu"
13
  # Load the processor from the local directory
14
+ processor = DonutProcessor.from_pretrained("Henge-navuuu/donut-base-finetuned-forms-v1")
15
 
16
  # Load the model from the local directory
17
+ model = VisionEncoderDecoderModel.from_pretrained("Henge-navuuu/donut-base-finetuned-forms-v1")
18
  model.to(device)
19
 
20
  @st.cache_resource