serenarolloh commited on
Commit
3eaed1b
·
verified ·
1 Parent(s): 173bbcf

Update routers/donut_inference.py

Browse files
Files changed (1) hide show
  1. routers/donut_inference.py +5 -1
routers/donut_inference.py CHANGED
@@ -18,7 +18,11 @@ def load_model():
18
  return processor, model, device
19
 
20
 
21
- def process_document_donut(image):
 
 
 
 
22
  worker_pid = os.getpid()
23
  print(f"Handling inference request with worker PID: {worker_pid}")
24
 
 
18
  return processor, model, device
19
 
20
 
21
+ def process_document_donut(image, settings):
22
+ processor = settings.processor
23
+ model = settings.model
24
+ # to load processor and model dynamically
25
+
26
  worker_pid = os.getpid()
27
  print(f"Handling inference request with worker PID: {worker_pid}")
28