ignore

s0md3v · Jul 15, 2023 · 9aa5023 · 9aa5023
1 parent f38239e
commit 9aa5023
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/roop/predictor.py b/roop/predictor.py
@@ -6,8 +6,20 @@
 
 from roop.typing import Frame
 
+PREDICTOR = None
+THREAD_LOCK = threading.Lock()
 MAX_PROBABILITY = 0.85
 
+
+def get_predictor() -> Model:
+    global PREDICTOR
+
+    with THREAD_LOCK:
+        if PREDICTOR is None:
+            PREDICTOR = opennsfw2.make_open_nsfw_model()
+    return PREDICTOR
+
+
 def clear_predictor() -> None:
     global PREDICTOR