Bobholamovic commited on
Commit
31204be
·
1 Parent(s): 448979e

Global model initialization lock

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -18,6 +18,8 @@ LANG_CONFIG = {
18
  }
19
  CONCURRENCY_LIMIT = 8
20
 
 
 
21
 
22
  class PaddleOCRModelManager(object):
23
  def __init__(self,
@@ -26,7 +28,6 @@ class PaddleOCRModelManager(object):
26
  super().__init__()
27
  self._model_factory = model_factory
28
  self._queue = Queue()
29
- self._model_init_lock = Lock()
30
  self._workers = []
31
  for _ in range(num_workers):
32
  worker = Thread(target=self._worker, daemon=False)
@@ -50,7 +51,7 @@ class PaddleOCRModelManager(object):
50
  worker.join()
51
 
52
  def _worker(self):
53
- with self._model_init_lock:
54
  model = self._model_factory()
55
  while True:
56
  item = self._queue.get()
 
18
  }
19
  CONCURRENCY_LIMIT = 8
20
 
21
+ model_init_lock = Lock()
22
+
23
 
24
  class PaddleOCRModelManager(object):
25
  def __init__(self,
 
28
  super().__init__()
29
  self._model_factory = model_factory
30
  self._queue = Queue()
 
31
  self._workers = []
32
  for _ in range(num_workers):
33
  worker = Thread(target=self._worker, daemon=False)
 
51
  worker.join()
52
 
53
  def _worker(self):
54
+ with model_init_lock:
55
  model = self._model_factory()
56
  while True:
57
  item = self._queue.get()