Spaces:
Runtime error
Runtime error
Commit ·
ab39bab
1
Parent(s): 805c468
Update app.py
Browse files
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 |
|