alexnasa commited on
Commit
5b42830
·
verified ·
1 Parent(s): 6156b3d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -6,6 +6,10 @@ import subprocess
6
  import uuid
7
  import shutil
8
  from tqdm import tqdm
 
 
 
 
9
 
10
 
11
  from huggingface_hub import snapshot_download, list_repo_files, hf_hub_download
@@ -291,6 +295,7 @@ def run_pipeline(prompt_text, steps, image_paths, audio_file_path, max_duration
291
  tea_cache_l1_thresh = 0.0,
292
  progress_bar_cmd=progress
293
  )
 
294
 
295
  # Return resulting video path
296
  video_path = os.path.join(output_dir, f"{filename}.mp4")
 
6
  import uuid
7
  import shutil
8
  from tqdm import tqdm
9
+ import threading
10
+
11
+ def print_thread_count(tag=""):
12
+ print(f"[DEBUG] {tag} Thread count:", len(threading.enumerate()))
13
 
14
 
15
  from huggingface_hub import snapshot_download, list_repo_files, hf_hub_download
 
295
  tea_cache_l1_thresh = 0.0,
296
  progress_bar_cmd=progress
297
  )
298
+ print_thread_count("after generation")
299
 
300
  # Return resulting video path
301
  video_path = os.path.join(output_dir, f"{filename}.mp4")