reab5555 commited on
Commit
e463715
·
verified ·
1 Parent(s): 392b31b

Update video_processing.py

Browse files
Files changed (1) hide show
  1. 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
- heatmap_video_path, original_fps, largest_cluster,
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