Add more batchsize

app.py

@@ -68,7 +68,7 @@ def main():
     
 
     with st.spinner(f'Processing {file_type}...'):
-        processing_stdout = subprocess.run(shlex.split(f"""python extract_video.py --device cuda --max_frames 50 --bs 32 --frame_interval 60 --confidence_threshold 0.997 --data_path "{filepath}" """), capture_output=True)
+        processing_stdout = subprocess.run(shlex.split(f"""python extract_video.py --device cuda --max_frames 50 --bs 128 --frame_interval 60 --confidence_threshold 0.997 --data_path "{filepath}" """), capture_output=True)
     st.text(f'1. Processing {file_type} ✅')
     with st.spinner(f'Analyzing {file_type}...'):
         analyze_stdout = subprocess.run(shlex.split(f"""python inference.py --weight weights/model_params_ffpp_c23.pickle --device cuda --image_folder "{base_folder}/images/{filename}" """), capture_output=True)