Spaces:
Runtime error
Runtime error
Update video_processing.py
Browse files- video_processing.py +2 -1
video_processing.py
CHANGED
|
@@ -214,9 +214,10 @@ def process_video(video_path, anomaly_threshold, desired_fps, progress=None):
|
|
| 214 |
if progress is not None:
|
| 215 |
progress(0.95, desc="Generating video with heatmap")
|
| 216 |
|
|
|
|
| 217 |
heatmap_video_path = create_video_with_heatmap(
|
| 218 |
video_path, df, mse_embeddings, mse_posture, mse_voice,
|
| 219 |
-
|
| 220 |
progress=progress
|
| 221 |
)
|
| 222 |
|
|
|
|
| 214 |
if progress is not None:
|
| 215 |
progress(0.95, desc="Generating video with heatmap")
|
| 216 |
|
| 217 |
+
output_folder = "output" # You can change this to your desired output folder
|
| 218 |
heatmap_video_path = create_video_with_heatmap(
|
| 219 |
video_path, df, mse_embeddings, mse_posture, mse_voice,
|
| 220 |
+
output_folder, original_fps, largest_cluster,
|
| 221 |
progress=progress
|
| 222 |
)
|
| 223 |
|