ManishThota commited on
Commit
12f87dd
·
verified ·
1 Parent(s): c8c4b3b

Update multi_video_app.py

Browse files
Files changed (1) hide show
  1. multi_video_app.py +2 -0
multi_video_app.py CHANGED
@@ -4,6 +4,7 @@ import gradio as gr
4
  import re
5
  from typing import Dict, List
6
  import os
 
7
  import torch
8
  import pandas as pd
9
  from src.video_model import describe_video
@@ -85,6 +86,7 @@ def process_multiple_videos(video_files: List[str], standing, hands, location, s
85
 
86
  # Clear GPU cache
87
  torch.cuda.empty_cache()
 
88
 
89
  # Save all observations to a CSV file and return the file path
90
  csv_file = save_to_csv(all_observations)
 
4
  import re
5
  from typing import Dict, List
6
  import os
7
+ import gc
8
  import torch
9
  import pandas as pd
10
  from src.video_model import describe_video
 
86
 
87
  # Clear GPU cache
88
  torch.cuda.empty_cache()
89
+ gc.collect()
90
 
91
  # Save all observations to a CSV file and return the file path
92
  csv_file = save_to_csv(all_observations)