diegulio commited on
Commit
98952d2
·
1 Parent(s): aa401bb

demo update

Browse files
Files changed (2) hide show
  1. cedula/app.py +2 -2
  2. license/app.py +2 -2
cedula/app.py CHANGED
@@ -99,8 +99,8 @@ def crop(files = '', #files
99
  def get_attributes(input_img):
100
  #access_token = str(os.environ.get('key'))
101
  access_token = key
102
- processor = DonutProcessor.from_pretrained("ClipAI/cedula-demo", use_auth_token=access_token)
103
- model = VisionEncoderDecoderModel.from_pretrained("ClipAI/cedula-demo", use_auth_token=access_token)
104
 
105
  device = "cuda" if torch.cuda.is_available() else "cpu"
106
 
 
99
  def get_attributes(input_img):
100
  #access_token = str(os.environ.get('key'))
101
  access_token = key
102
+ processor = DonutProcessor.from_pretrained("ClipAI/donut-finetuned-crop-cedula", use_auth_token=access_token)
103
+ model = VisionEncoderDecoderModel.from_pretrained("ClipAI/donut-finetuned-crop-cedula", use_auth_token=access_token)
104
 
105
  device = "cuda" if torch.cuda.is_available() else "cpu"
106
 
license/app.py CHANGED
@@ -99,8 +99,8 @@ def crop(files = '', #files
99
  def get_attributes(input_img):
100
  #access_token = str(os.environ.get('key'))
101
  access_token = key
102
- processor = DonutProcessor.from_pretrained("ClipAI/license-demo", use_auth_token=access_token)
103
- model = VisionEncoderDecoderModel.from_pretrained("ClipAI/license-demo", use_auth_token=access_token)
104
 
105
  device = "cuda" if torch.cuda.is_available() else "cpu"
106
 
 
99
  def get_attributes(input_img):
100
  #access_token = str(os.environ.get('key'))
101
  access_token = key
102
+ processor = DonutProcessor.from_pretrained("ClipAI/donut-finetuned-crop-licencia", use_auth_token=access_token)
103
+ model = VisionEncoderDecoderModel.from_pretrained("ClipAI/donut-finetuned-crop-licencia", use_auth_token=access_token)
104
 
105
  device = "cuda" if torch.cuda.is_available() else "cpu"
106