log duration

app.py

@@ -1,5 +1,6 @@
 import os
 import time
+import logging
 from pathlib import Path
 
 import torch
@@ -17,6 +18,7 @@ from esrgan_model import UpscalerESRGAN
 from model import create_model
 
 device = "cuda"
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(message)s', datefmt="%Y-%m-%d %H:%M:%S")
 
 # Custom transform to pad images to square
 class PadToSquare:
@@ -33,7 +35,7 @@ def timer_func(func):
     def wrapper(*args, **kwargs):
         t0 = time.time()
         result = func(*args, **kwargs)
-        print(f"{func.__name__} took {time.time() - t0:.2f} seconds")
+        logging.info(f"{func.__name__} took {time.time() - t0:.2f} seconds")
         return result
     return wrapper