mrneuralnet commited on
Commit
83192f3
·
1 Parent(s): db1fe34

Add setup function

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -40,7 +40,7 @@ def main():
40
 
41
 
42
  with st.spinner(f'Processing {file_type}...'):
43
- subprocess.run(shlex.split(f"python.exe extract_video.py --device cuda --max_frames 50 --bs 2 --frame_interval 5 --confidence_threshold 0.997 --data_path temps/{filename}"))
44
  st.text(f'1. Processing {file_type} ✅')
45
  with st.spinner(f'Analyzing {file_type}...'):
46
  pred = subprocess.run(shlex.split(f"python inference.py --weight weights/model_params_ffpp_c23.pickle --device cuda --image_folder temps/images/{filename}"), capture_output=True)
 
40
 
41
 
42
  with st.spinner(f'Processing {file_type}...'):
43
+ subprocess.run(shlex.split(f"python extract_video.py --device cuda --max_frames 50 --bs 2 --frame_interval 5 --confidence_threshold 0.997 --data_path temps/{filename}"))
44
  st.text(f'1. Processing {file_type} ✅')
45
  with st.spinner(f'Analyzing {file_type}...'):
46
  pred = subprocess.run(shlex.split(f"python inference.py --weight weights/model_params_ffpp_c23.pickle --device cuda --image_folder temps/images/{filename}"), capture_output=True)