serenarolloh commited on
Commit
51e9d6e
·
verified ·
1 Parent(s): e26f64e

Update routers/inference.py

Browse files
Files changed (1) hide show
  1. routers/inference.py +6 -3
routers/inference.py CHANGED
@@ -28,9 +28,12 @@ def count_values(obj):
28
 
29
 
30
  @router.post("/inference")
31
- async def run_inference(file: Optional[UploadFile] = File(None), image_url: Optional[str] = Form(None),
32
- model_in_use: str = Form('donut')):
33
-
 
 
 
34
 
35
  result = []
36
  if file:
 
28
 
29
 
30
  @router.post("/inference")
31
+ async def run_inference(
32
+ file: Optional[UploadFile] = File(None),
33
+ image_url: Optional[str] = Form(None),
34
+ shipper_id: str = Form(...),
35
+ model_in_use: Optional[str] = Form(None)
36
+ ):
37
 
38
  result = []
39
  if file: