Update app.py

app.py

@@ -31,7 +31,7 @@ from moviepy.editor import VideoFileClip, AudioFileClip
 import multiprocessing
 import imageio
 import tqdm
-
+from concurrent.futures import ProcessPoolExecutor
 
 subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
 
@@ -531,7 +531,8 @@ def generate_video(image, prompt, max_tokens):
         heatmap_imgs.append(overlay)
     
     # Dùng multiprocessing
-    with multiprocessing.Pool(processes=20) as pool:
+    # with multiprocessing.Pool(processes=20) as pool:
+    with ProcessPoolExecutor(max_workers=20) as pool:
         text_imgs = pool.map(generate_text_img_wrapper, params_for_text)
         hidden_imgs = pool.map(generate_hidden_img_wrapper, params_for_hidden)