Update run.py

run.py

@@ -6,19 +6,19 @@ from insightface.app import FaceAnalysis
 import tempfile
 from moviepy.editor import VideoFileClip
 
-# ===== Configuration précoce pour éviter les erreurs =====
-
+# Désactiver les warnings d'Albumentations
 os.environ["NO_ALBUMENTATIONS_UPDATE"] = "1"
+
+# Rediriger les chemins problématiques
 os.environ['MPLCONFIGDIR'] = '/tmp/matplotlib'
 os.environ['FONTCONFIG_PATH'] = '/tmp/fontconfig'
 
 os.makedirs('/tmp/matplotlib', exist_ok=True)
 os.makedirs('/tmp/fontconfig', exist_ok=True)
 
+# Forcer InsightFace à utiliser un répertoire accessible
 os.environ['INSIGHTFACE_ROOT'] = '/tmp/.insightface'
-os.environ["ORT_DISABLE_CUDA"] = "1"
-
-# ===== Fonction de swap =====
+os.environ["ORT_DISABLE_CUDA"] = "1"  # Désactiver CUDA si GPU indisponible
 
 def swap_face(source_face, target_face, frame):
     src_emb = source_face.normed_embedding
@@ -78,19 +78,18 @@ def process_video(source_img, target_video):
         return final_path
 
     except Exception as e:
-        print(f"Erreur : {str(e)}")
+        print(f"Erreur lors du traitement : {str(e)}")
         return None
 
 
-# ===== Interface Gradio =====
-
+# Interface Gradio
 demo = gr.Interface(
     fn=process_video,
     inputs=[
-        gr.Image(label="Visage Source", type="numpy"),  # ✅ Correct
-        gr.Video(label="Vidéo Cible"),                  # ✅ Correct
+        gr.Image(label="Visage Source", type="numpy"),
+        gr.Video(label="Vidéo Cible"),
     ],
-    outputs=gr.Video(label="Vidéo Résultat"),           # ✅ Correct
+    outputs=gr.Video(label="Vidéo Résultat"),
     title="🎬 FaceSwap Pro",
     description="Échangez des visages dans une vidéo.",
     allow_flagging="never"