More time analysis

Taxabind/TaxaBind/SatBind/clip_seg_tta.py

@@ -438,7 +438,9 @@ class ClipSegTTA:
                 )
 
         # Save final heatmap after TTA steps 
+        time_start = time.time()
         self.generate_heatmap(img, imo, txt, sound=sound, modality=modality)
+        print(f"Time taken to generate heatmap after TTA steps:  {(time.time()-time_start):.2f}")
 
         ## NOTE: Added due to app.py (to allocate to GPU only when needed on HF)
         # if self.device.type == "cuda":