mrneuralnet commited on
Commit
c759fb9
1 Parent(s): 51b1aed
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -43,7 +43,7 @@ def main():
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)
47
  st.text(f'2. Analyzing {file_type} ✅')
48
 
49
  print(pred)
 
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 cpu --image_folder temps/images/{filename}"), capture_output=True)
47
  st.text(f'2. Analyzing {file_type} ✅')
48
 
49
  print(pred)