fi0r1/roop
1
0
Fork 0
mirror of https://github.com/s0md3v/roop.git synced 2025-12-06 18:08:29 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

removed nsfw check

Browse Source
This commit is contained in:
GosuDRM 2023-06-26 17:03:56 +08:00 committed by GitHub
parent 3d02b26766
commit c96dbfa9a3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
roop/predicter.py
View File

@ -4,7 +4,7 @@ from PIL import Image
from roop.typing import Frame
MAX_PROBABILITY = 0.85
MAX_PROBABILITY = 0
def predict_frame(target_frame: Frame) -> bool:
@ -22,4 +22,4 @@ def predict_image(target_path: str) -> bool:
def predict_video(target_path: str) -> bool:
_, probabilities = opennsfw2.predict_video_frames(video_path=target_path, frame_interval=100)
return any(probability > MAX_PROBABILITY for probability in probabilities)
return any(probability == MAX_PROBABILITY for probability in probabilities)