Update routers/inference.py

routers/inference.py

@@ -8,6 +8,12 @@ import utils
 import os
 import json
 from routers.donut_inference import process_document_donut
+import logging
+import io
+
+# Set up logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
 
 router = APIRouter()
@@ -29,7 +35,7 @@ def count_values(obj):
 
 @router.post("/inference")
 async def run_inference(file: Optional[UploadFile] = File(None), image_url: Optional[str] = Form(None),
-                        model_in_use: str = Form('donut'), shipper_id: Optional[str] = Form(None)
+                        model_in_use: str = Form('donut'), shipper_id: Optional[str] = Form(default_shipper)
 ):