From 1aae8266c22d8f69eddc03c744de07c6b6688e46 Mon Sep 17 00:00:00 2001
From: Nick Mazurkin <mazurkin@gmail.com>
Date: Wed, 25 Sep 2024 13:00:22 -0400
Subject: [PATCH] environment variable PROBABILITY_LIMIT

---
 facefusion/content_analyser.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/facefusion/content_analyser.py b/facefusion/content_analyser.py
index 998e560c..e42d4107 100644
--- a/facefusion/content_analyser.py
+++ b/facefusion/content_analyser.py
@@ -2,6 +2,7 @@ from functools import lru_cache
 
 import cv2
 import numpy
+import os
 from tqdm import tqdm
 
 from facefusion import inference_manager, state_manager, wording
@@ -35,7 +36,7 @@ MODEL_SET : ModelSet =\
 		'mean': [ 104, 117, 123 ]
 	}
 }
-PROBABILITY_LIMIT = 0.80
+PROBABILITY_LIMIT = os.environ.get('FACEFUSION_PROBABILITY_LIMIT', 0.80)
 RATE_LIMIT = 10
 STREAM_COUNTER = 0