Update face_swap.py

face_swap.py

@@ -17,11 +17,11 @@ os.makedirs(os.environ["MPLCONFIGDIR"], exist_ok=True)
 
 # Try GPU, fallback to CPU
 try:
-    face_analyzer = FaceAnalysis(name='buffalo_l', providers=['CUDAExecutionProvider'])
+    face_analyzer = FaceAnalysis(name='buffalo_l', root=cache_dir, providers=['CUDAExecutionProvider'])
     face_analyzer.prepare(ctx_id=0)
 except Exception:
     print("⚠️ CUDA not available for FaceAnalysis. Falling back to CPU.")
-    face_analyzer = FaceAnalysis(name='buffalo_l', providers=['CPUExecutionProvider'])
+    face_analyzer = FaceAnalysis(name='buffalo_l', root=cache_dir, providers=['CPUExecutionProvider'])
     face_analyzer.prepare(ctx_id=0)
 
 try: