JoseArcosO commited on
Commit
ab39bab
·
1 Parent(s): 805c468

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -94,8 +94,8 @@ def crop(files = '', #files
94
  def get_attributes(input_img):
95
  #access_token = str(os.environ.get('key'))
96
  access_token = key
97
- processor = DonutProcessor.from_pretrained("AA-supply/donut-finetuned-cedula", use_auth_token=access_token)
98
- model = VisionEncoderDecoderModel.from_pretrained("AA-supply/donut-finetuned-cedula", use_auth_token=access_token)
99
 
100
  device = "cuda" if torch.cuda.is_available() else "cpu"
101
 
 
94
  def get_attributes(input_img):
95
  #access_token = str(os.environ.get('key'))
96
  access_token = key
97
+ processor = DonutProcessor.from_pretrained("AA-supply/donut-finetuned-cedula-crop", use_auth_token=access_token)
98
+ model = VisionEncoderDecoderModel.from_pretrained("AA-supply/donut-finetuned-cedula-crop", use_auth_token=access_token)
99
 
100
  device = "cuda" if torch.cuda.is_available() else "cpu"
101