Update app.py

app.py

@@ -3,6 +3,8 @@ import spaces
 import torch
 from PIL import Image
 from RealESRGAN import RealESRGAN
+import time
+from datetime import timedelta as td
 
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 model2 = RealESRGAN(device, scale=2)
@@ -15,6 +17,7 @@ model8.load_weights('weights/RealESRGAN_x8.pth', download=True)
 
 @spaces.GPU(duration=14)
 def inference(image, size):
+    start_load = time.time()
     global model2
     global model4
     global model8
@@ -50,7 +53,7 @@ def inference(image, size):
             model8.load_weights('weights/RealESRGAN_x8.pth', download=False)
             result = model2.predict(image.convert('RGB'))
             
-    print(f"Image size ({device}): {size} ... OK")
+    print(f"Image size ({device}): {size}, time: {datetime.datetime.now()} ... OK")
     return result
 
 
@@ -73,4 +76,4 @@ gr.Interface(inference,
     flagging_mode="never",
     cache_mode="lazy",
     delete_cache=(44000, 44000),
-    ).queue(api_open=True).launch(show_error=True, show_api=True, mcp_server=True)
+    ).queue(api_open=True).launch(show_error=True, show_api=True, mcp_server=True)