Update app.py

app.py

@@ -8,6 +8,8 @@ import numpy as np
 import spaces
 import torch
 import random
+import time
+from datetime import timedelta as td
 from PIL import Image
 
 from diffusers import FluxKontextPipeline
@@ -22,6 +24,7 @@ optimize_pipeline_(pipe, image=Image.new("RGB", (512, 512)), prompt='prompt')
 
 @spaces.GPU(duration=24)
 def infer(input_image, prompt, seed=42, randomize_seed=False, guidance_scale=2.5, steps=28, progress=gr.Progress(track_tqdm=True)):
+    start_load = time.time()
     """
     Perform image editing using the FLUX.1 Kontext pipeline.
     
@@ -82,11 +85,15 @@ def infer(input_image, prompt, seed=42, randomize_seed=False, guidance_scale=2.5
             num_inference_steps=steps,
             generator=torch.Generator().manual_seed(seed),
         ).images[0]
+    
+    print(f"Time Elapsed: {td(seconds=int(time.time() - start_load))}")
     return image, seed, gr.Button(visible=True)
 
 @spaces.GPU(duration=25)
 def infer_example(input_image, prompt):
+    start_load = time.time()
     image, seed, _ = infer(input_image, prompt)
+    print(f"Time Elapsed: {td(seconds=int(time.time() - start_load))}")
     return image, seed
 
 css="""