Spaces:
Runtime error
Runtime error
Update multi_video_app.py
Browse files- 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)
|