invoice_document_headers_extraction_with_donut

Paused

to-be commited on Mar 3, 2023

Commit

a773f5d

1 Parent(s): 4fd7d4e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,8 +10,8 @@ import json
 import os
-processor = DonutProcessor.from_pretrained("./donut-base-finetuned-inv")
-model = VisionEncoderDecoderModel.from_pretrained("./donut-base-finetuned-inv")
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model.to(device)

 import os
+processor = DonutProcessor.from_pretrained("to-be/donut-base-finetuned-invoices")
+model = VisionEncoderDecoderModel.from_pretrained("to-be/donut-base-finetuned-invoices")
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model.to(device)