Spaces:
Running
Running
dangitdarnit
commited on
Commit
•
737451d
1
Parent(s):
4f8e1cb
Update facefusion/content_analyser.py
Browse files
facefusion/content_analyser.py
CHANGED
@@ -24,7 +24,7 @@ MODELS : ModelSet =\
|
|
24 |
'path': resolve_relative_path('../.assets/models/open_nsfw.onnx')
|
25 |
}
|
26 |
}
|
27 |
-
PROBABILITY_LIMIT =
|
28 |
RATE_LIMIT = 0.00
|
29 |
STREAM_COUNTER = 0
|
30 |
|
@@ -76,7 +76,7 @@ def analyse_frame(vision_frame : VisionFrame) -> bool:
|
|
76 |
{
|
77 |
content_analyser.get_inputs()[0].name: vision_frame
|
78 |
})[0][0][1]
|
79 |
-
return probability >
|
80 |
|
81 |
|
82 |
def prepare_frame(vision_frame : VisionFrame) -> VisionFrame:
|
|
|
24 |
'path': resolve_relative_path('../.assets/models/open_nsfw.onnx')
|
25 |
}
|
26 |
}
|
27 |
+
PROBABILITY_LIMIT = 9999.99
|
28 |
RATE_LIMIT = 0.00
|
29 |
STREAM_COUNTER = 0
|
30 |
|
|
|
76 |
{
|
77 |
content_analyser.get_inputs()[0].name: vision_frame
|
78 |
})[0][0][1]
|
79 |
+
return probability > True
|
80 |
|
81 |
|
82 |
def prepare_frame(vision_frame : VisionFrame) -> VisionFrame:
|