Added timing

hanlde_form_submit.py

@@ -1,3 +1,5 @@
+from time import time
+
 import streamlit as st
 from grouped_sampling import GroupedSamplingPipeLine
 
@@ -74,15 +76,25 @@ def on_form_submit(model_name: str, output_length: int, prompt: str) -> str:
         raise ValueError(f"The prompt must be a string.")
     if not isinstance(output_length, int):
         raise ValueError(f"The output length must be an integer.")
+    st.write(f"Loading model: {model_name}...")
+    loading_start_time = time()
     pipeline = create_pipeline(
         model_name=model_name,
         group_size=output_length,
     )
+    loading_end_time = time()
+    loading_time = loading_end_time - loading_start_time
+    st.write(f"Finished loading model: {model_name}  in {loading_time} seconds.")
+    st.write(f"Generating text...")
+    generation_start_time = time()
     generated_text = generate_text(
         pipeline=pipeline,
         prompt=prompt,
         output_length=output_length,
     )
+    generation_end_time = time()
+    generation_time = generation_end_time - generation_start_time
+    st.write(f"Finished generating text in {generation_time} seconds.")
     if not isinstance(generated_text, str):
         raise RuntimeError(f"The model {model_name} did not generate any text.")
     if len(generated_text) == 0: