serenarolloh commited on
Commit
0ca384b
·
verified ·
1 Parent(s): 3940dee

Update routers/donut_inference.py

Browse files
Files changed (1) hide show
  1. routers/donut_inference.py +1 -1
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