update logger

app.py

@@ -18,7 +18,13 @@ 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 timer logger only
+timer_logger = logging.getLogger("TIMER")
+timer_logger.setLevel(logging.INFO)
+handler = logging.StreamHandler()  # Attach a stream handler with formatter
+handler.setFormatter(logging.Formatter('%(asctime)s - %(message)s', datefmt="%Y-%m-%d %H:%M:%S"))
+timer_logger.addHandler(handler)
+timer_logger.propagate = False  # Avoid duplicate logs
 
 # Custom transform to pad images to square
 class PadToSquare:
@@ -35,7 +41,7 @@ def timer_func(func):
     def wrapper(*args, **kwargs):
         t0 = time.time()
         result = func(*args, **kwargs)
-        logging.info(f"{func.__name__} took {time.time() - t0:.2f} seconds")
+        timer_logger.info(f"{func.__name__} took {time.time() - t0:.2f} seconds")
         return result
     return wrapper