finding bottlenecks pt1

interfaces/manifesto.py

@@ -11,6 +11,9 @@ from huggingface_hub import HfApi
 from label_dicts import MANIFESTO_LABEL_NAMES
 
 class RuntimeMeasure:
+    def __init__(self, msg):
+        self.msg = msg
+
     def __enter__(self):
         self.start_time = time.time()
         return self
@@ -18,7 +21,7 @@ class RuntimeMeasure:
     def __exit__(self, exc_type, exc_value, traceback):
         end_time = time.time()
         runtime = end_time - self.start_time
-        gr.Info(f"Runtime: {runtime} seconds")
+        gr.Info(f"{self.msg}: {runtime} seconds")
 def m(msg):
     return RuntimeMeasure(msg)