Add slider

app.py

@@ -91,13 +91,14 @@ if __name__ == "__main__":
     image_pipeline = FaceFakePipelineImage()
     video_pipeline = FaceFakePipelineVideo()
 
-    config_payload = {
-        'face_detection_threshold': 0.997,
-        'deformation_detection_threshold': 0.6,
-        'deepfake_detection_threshold': 0.65,
-        'frame_sampling_interval': 60,
-        'frame_sampling_max': 50,
-    }
+    with st.sidebar:       
+        config_payload = {
+            'face_detection_threshold': 0.997,
+            'deformation_detection_threshold': 0.6,
+            'deepfake_detection_threshold': 0.65,
+            'frame_sampling_interval': st.slider('frame_sampling_interval', 0, 60, 300, step=5),
+            'frame_sampling_max': st.slider('frame_sampling_max', 0, 50, 100),
+        }
 
     st.set_page_config(
         page_title="Nodeflux Face Fake Detection", page_icon=":pencil2:"

deformation_detection/main.py

@@ -21,7 +21,7 @@ from deformation_detection.utils.visualize import *
 from deformation_detection.utils.tools import resize_shorter_side
 
 class DeformationDetection:
-    def __init__(self, batch_size=16, device=None, gpu_id=0, global_weights='weights/global.pth', local_weights='weights\local.pth'):
+    def __init__(self, batch_size=16, device=None, gpu_id=0, global_weights='weights/global.pth', local_weights='weights/local.pth'):
         if device:
             self.device = device
         elif torch.cuda.is_available() and gpu_id != -1: