Update app.py

app.py

@@ -1,8 +1,12 @@
 import gradio as gr
 import torch
-from models.TCVC import TCVC_Model  # Assurez-vous que le chemin est correct après avoir cloné le repo
-import cv2
 import os
+from TCVC-Temporally-Consistent-Video-Colorization.models.TCVC import TCVC_Model  # Charger le modèle depuis le bon chemin
+import cv2
+
+# Clone le repo si non présent
+if not os.path.exists('TCVC-Temporally-Consistent-Video-Colorization'):
+    os.system('git clone https://github.com/lyh-18/TCVC-Temporally-Consistent-Video-Colorization.git')
 
 # Charger le modèle pré-entraîné
 def load_model():
@@ -48,6 +52,5 @@ interface = gr.Interface(fn=process_video,
                          title="Colorisation vidéo",
                          description="Téléchargez une vidéo en noir et blanc et obtenez une version colorisée.")
 
-# Lancer l'interface
 if __name__ == "__main__":
     interface.launch()