Update app.py

app.py

@@ -40,13 +40,11 @@ def frames_to_video(frame_folder, video_path, image_path, frame_count,fps):
 
     fourcc = cv2.VideoWriter_fourcc(*'mp4v')
     out = cv2.VideoWriter(video_path, fourcc, fps, (width, height))
-    
-    providers = ["CUDAExecutionProvider"]
-    print("Available Providers==",providers)
-    app = insightface.app.FaceAnalysis(name='buffalo_l', providers=providers)
-    app.prepare(ctx_id=0, det_size=(640, 640))
-    swapper = insightface.model_zoo.get_model("inswapper_128.onnx",download=False, download_zip=False,providers=providers)
-    face_enhancer = gfpgan.GFPGANer(model_path="GFPGANv1.4.pth", upscale=1, device='cuda')
+    num_gpus = 3  # Adjust based on the number of GPUs you have
+    providers = [f"CUDAExecutionProvider:{i}" for i in range(num_gpus)]
+    apps = [insightface.app.FaceAnalysis(name='buffalo_l', providers=[provider]) for provider in providers]
+    swappers = [insightface.model_zoo.get_model("inswapper_128.onnx", download=False, download_zip=False, providers=[provider]) for provider in providers]
+    face_enhancers = [gfpgan.GFPGANer(model_path="GFPGANv1.4.pth", upscale=1, device=f'cuda:{i}') for i in range(num_gpus)]
     
     for i in tqdm(range(frame_count), desc="Converting frames to video"):
         print("Progress:",progress)