Change to cpu

app.py

@@ -68,10 +68,10 @@ 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 128 --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 cpu --max_frames 50 --bs 2 --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)
+        analyze_stdout = subprocess.run(shlex.split(f"""python inference.py --weight weights/model_params_ffpp_c23.pickle --device cpu --image_folder "{base_folder}/images/{filename}" """), capture_output=True)
     st.text(f'2. Analyzing {file_type} ✅')
 
     if len(os.listdir("{}/images/{}".format(base_folder, filename.split(".")[0]))) < 1: