Spaces:
Running
Running
Bobholamovic
commited on
Commit
·
31204be
1
Parent(s):
448979e
Global model initialization lock
Browse files
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
|
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()
|