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()
|