ivelin commited on
Commit
94fd60d
·
1 Parent(s): 0eab7dd

fix: move hf token to repo envs

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -23,7 +23,7 @@ def load_model(pretrained_revision: str = 'main'):
23
  if processor is None or loaded_revision is None or loaded_revision != pretrained_revision:
24
  loaded_revision=pretrained_revision
25
  processor = DonutProcessor.from_pretrained(
26
- pretrained_repo_name, revision=pretrained_revision, use_auth_token="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")
27
  processor.image_processor.do_align_long_axis = False
28
  # do not manipulate image size and position
29
  processor.image_processor.do_resize = False
@@ -33,7 +33,7 @@ def load_model(pretrained_revision: str = 'main'):
33
  processor.image_processor.do_normalize = True
34
  print(f'processor image size: {processor.image_processor.size}')
35
  model = VisionEncoderDecoderModel.from_pretrained(
36
- pretrained_repo_name, use_auth_token="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", revision=pretrained_revision)
37
  print(f'model checkpoint loaded')
38
  device = "cuda" if torch.cuda.is_available() else "cpu"
39
  model.to(device)
 
23
  if processor is None or loaded_revision is None or loaded_revision != pretrained_revision:
24
  loaded_revision=pretrained_revision
25
  processor = DonutProcessor.from_pretrained(
26
+ pretrained_repo_name, revision=pretrained_revision) # , use_auth_token="...")
27
  processor.image_processor.do_align_long_axis = False
28
  # do not manipulate image size and position
29
  processor.image_processor.do_resize = False
 
33
  processor.image_processor.do_normalize = True
34
  print(f'processor image size: {processor.image_processor.size}')
35
  model = VisionEncoderDecoderModel.from_pretrained(
36
+ pretrained_repo_name, revision=pretrained_revision) # use_auth_token="...",
37
  print(f'model checkpoint loaded')
38
  device = "cuda" if torch.cuda.is_available() else "cpu"
39
  model.to(device)