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 |
|