Add timestamp

app.py

@@ -3,6 +3,7 @@ import torch
 import process_stylization
 from photo_wct import PhotoWCT
 import gradio as gr
+from datetime import datetime
 
 # Load model
 model_path = './models/photo_wct.pth'
@@ -10,6 +11,10 @@ p_wct = PhotoWCT()
 p_wct.load_state_dict(torch.load(model_path))
 
 def run(content_img, style_img, cuda, post_processing, fast):
+    now = datetime.now() 
+    dt_string = now.strftime("%d/%m/%Y %H:%M:%S")
+    print("[TimeStamp] {}".format(dt_string))
+    
     if fast == 0:
         from photo_gif import GIFSmoothing
         p_pro = GIFSmoothing(r=35, eps=0.001)
@@ -28,7 +33,7 @@ def run(content_img, style_img, cuda, post_processing, fast):
         cuda=cuda,
         post_processing=post_processing
     )
-
+    
     return output_img
 
 if __name__ == '__main__':