daquarti commited on
Commit
c54a77d
·
1 Parent(s): 65e8bf8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -7,9 +7,11 @@ import random
7
  import numpy as np
8
  import gradio as gr
9
 
 
 
10
  transformers.logging.disable_default_handler()
11
- processor = DonutProcessor.from_pretrained("daquarti/donut-base-sroie")
12
- model = VisionEncoderDecoderModel.from_pretrained("daquarti/donut-base-sroie")
13
  device = "cuda" if torch.cuda.is_available() else "cpu"
14
  model.to(device)
15
 
 
7
  import numpy as np
8
  import gradio as gr
9
 
10
+ access_token = "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"
11
+
12
  transformers.logging.disable_default_handler()
13
+ processor = DonutProcessor.from_pretrained("daquarti/donut-base-sroie", use_auth_token=access_token)
14
+ model = VisionEncoderDecoderModel.from_pretrained("daquarti/donut-base-sroie", use_auth_token=access_token)
15
  device = "cuda" if torch.cuda.is_available() else "cpu"
16
  model.to(device)
17