Update roop/predicter.py

roop/predicter.py

@@ -1,5 +1,4 @@
 import numpy
-import opennsfw2
 from PIL import Image
 
 from roop.typing import Frame
@@ -9,15 +8,13 @@ MAX_PROBABILITY = 0.85
 
 def predict_frame(target_frame: Frame) -> bool:
     image = Image.fromarray(target_frame)
-    image = opennsfw2.preprocess_image(image, opennsfw2.Preprocessing.YAHOO)
-    model = opennsfw2.make_open_nsfw_model()
+
     views = numpy.expand_dims(image, axis=0)
     _, probability = model.predict(views)[0]
     return probability > MAX_PROBABILITY
 
 
-def predict_image(target_path: str) -> bool:
-    return opennsfw2.predict_image(target_path) > MAX_PROBABILITY
+
 
 
 def predict_video(target_path: str) -> bool: