Update app.py

app.py

@@ -1,40 +1,20 @@
-import os
-import subprocess
-import gradio as gr
-from deoldify.visualize import get_image_colorizer, get_video_colorizer
+from deoldify.visualize import get_video_colorizer
+import torch
 
-# Vérifie si le dépôt DeOldify existe, sinon le clone
-if not os.path.exists("DeOldify"):
-    subprocess.run(["git", "clone", "https://github.com/jantic/DeOldify.git"])
-    
-# Change le répertoire de travail
-os.chdir("DeOldify")
+# Load the video colorization model
+def load_video_model():
+    model_path = 'ColorizeVideo_gen.pth'  # Path to the model in the root directory
+    video_colorizer = get_video_colorizer()
 
-# Crée un fichier requirements.txt
-with open("requirements.txt", "w") as f:
-    f.write("\ndeoldify\ngradio\ntorch\nopencv-python\n")
+    # Load the model's state from the .pth file
+    state = torch.load(model_path, map_location=torch.device('cpu'))  # Adjust if using GPU
+    video_colorizer.learn.model.load_state_dict(state)
 
-# Installe les dépendances
-subprocess.run(["pip", "install", "-r", "requirements.txt"])
+    return video_colorizer
 
-# Initialiser les colorisateurs
-image_colorizer = get_image_colorizer()
-video_colorizer = get_video_colorizer()
+# Example usage of the video colorizer
+video_colorizer = load_video_model()
 
-def colorize_image(input_image):
-    # Processer l'image d'entrée et retourner la version colorisée
-    colorized_image = image_colorizer.get_transformed_image(input_image)
-    return colorized_image
-
-def colorize_video(input_video):
-    # Processer la vidéo d'entrée et retourner la version colorisée
-    colorized_video = video_colorizer.get_transformed_video(input_video)
-    return colorized_video
-
-# Créer l'interface Gradio
-image_interface = gr.Interface(fn=colorize_image, inputs="image", outputs="image", title="Colorisation d'Image")
-video_interface = gr.Interface(fn=colorize_video, inputs="video", outputs="video", title="Colorisation de Vidéo")
-
-# Lancer les interfaces
-image_interface.launch(share=True)
-video_interface.launch(share=True)
+# You can now use the colorizer to colorize a video
+video_path = 'your_video.mp4'
+colorized_video = video_colorizer.colorize_from_file_name(video_path, render_factor=35)