Spaces:
Sleeping
Sleeping
Commit
·
83192f3
1
Parent(s):
db1fe34
Add setup function
Browse files
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
|
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)
|