Update colorization.py

colorization.py

@@ -1,8 +1,8 @@
+import subprocess
 import os
 import gradio as gr
 import paddlehub as hub
 from pathlib import Path
-from datetime import datetime
 
 # Install DeOldify model
 os.system("hub install deoldify==1.0.1")
@@ -28,6 +28,6 @@ def create_interface():
         description="Upload a black and white photo to colorize it.",
     ).launch()
 
-# Run the Gradio interface
 if __name__ == "__main__":
+    # Run the Gradio interface
     create_interface()