Update routers/inference.py

routers/inference.py

@@ -42,7 +42,7 @@ async def run_inference(file: Optional[UploadFile] = File(None), image_url: Opti
         image = Image.open(BytesIO(await file.read()))
         processing_time = 0
         if model_in_use == 'donut':
-            result, processing_time = process_document_donut(image)
+            result, processing_time = process_document_donut(image, shipper_id)
         utils.log_stats(settings.inference_stats_file, [processing_time, count_values(result), file.filename, settings.model])
         print(f"Processing time: {processing_time:.2f} seconds")
     elif image_url:
@@ -52,7 +52,7 @@ async def run_inference(file: Optional[UploadFile] = File(None), image_url: Opti
 
         processing_time = 0
         if model_in_use == 'donut':
-            result, processing_time = process_document_donut(image)
+            result, processing_time = process_document_donut(image, shipper_id)
         # parse file name from url
         file_name = image_url.split("/")[-1]
         utils.log_stats(settings.inference_stats_file, [processing_time, count_values(result), file_name, settings.model])