measure whole process

interfaces/manifesto.py

@@ -67,9 +67,11 @@ def predict(text, model_id, tokenizer_id):
     return output_pred, output_info
 
 def predict_cap(text, language):
-    model_id = build_huggingface_path(language)
-    tokenizer_id = "xlm-roberta-large"
-    return predict(text, model_id, tokenizer_id)
+    with m("WHOLE PROCESS"):
+        model_id = build_huggingface_path(language)
+        tokenizer_id = "xlm-roberta-large"
+        prediction = predict(text, model_id, tokenizer_id)
+    return prediction
 
 demo = gr.Interface(
     fn=predict_cap,