Spaces:
Sleeping
Sleeping
Update routers/donut_inference.py
Browse files
routers/donut_inference.py
CHANGED
@@ -60,7 +60,7 @@ def load_model(shipper_id: str):
|
|
60 |
model_cache['default_shipper'] = (processor, model, device)
|
61 |
return model_cache['default_shipper']
|
62 |
|
63 |
-
def process_document_donut(image, shipper_id):
|
64 |
"""
|
65 |
Process a document using the appropriate model for the shipper
|
66 |
|
|
|
60 |
model_cache['default_shipper'] = (processor, model, device)
|
61 |
return model_cache['default_shipper']
|
62 |
|
63 |
+
def process_document_donut(image, shipper_id="default_shipper"):
|
64 |
"""
|
65 |
Process a document using the appropriate model for the shipper
|
66 |
|