ignore

roop/predictor.py

@@ -6,8 +6,20 @@ from keras import Model
 
 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