Update app.py

app.py

@@ -1,18 +1,23 @@
-import gradio as gr
-import torch
 import os
 import sys
+import time
+import gradio as gr
+import torch
 import cv2
 
-# Clone le repo si non présent
+# Cloner 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')
 
+# Vérifier si le dépôt est bien cloné
+while not os.path.exists('TCVC-Temporally-Consistent-Video-Colorization'):
+    time.sleep(1)  # Attendre que le clonage soit terminé
+
 # Ajouter le dépôt cloné au chemin Python
 sys.path.append('TCVC-Temporally-Consistent-Video-Colorization')
 
-# Importer le modèle depuis le dépôt cloné
-from models.TCVC import TCVC_Model
+# Importer le modèle une fois le dépôt cloné
+from TCVC_Temporally_Consistent_Video_Colorization.models.TCVC import TCVC_Model
 
 # Charger le modèle pré-entraîné
 def load_model():