added print statements

hanlde_form_submit.py

@@ -88,6 +88,7 @@ def on_form_submit(
     if not is_supported(model_name):
         raise UnsupportedModelNameException(model_name)
     st.write(f"Loading model: {model_name}...")
+    print(f"Loading model: {model_name}...")
     loading_start_time = time()
     pipeline = create_pipeline(
         model_name=model_name,
@@ -96,7 +97,9 @@ def on_form_submit(
     loading_end_time = time()
     loading_time = loading_end_time - loading_start_time
     st.write(f"Finished loading model: {model_name}  in {loading_time:,.2f} seconds.")
+    print(f"Finished loading model: {model_name}  in {loading_time:,} seconds.")
     st.write("Generating text...")
+    print("Generating text...")
     generation_start_time = time()
     generated_text = generate_text(
         pipeline=pipeline,
@@ -106,6 +109,7 @@ def on_form_submit(
     generation_end_time = time()
     generation_time = generation_end_time - generation_start_time
     st.write(f"Finished generating text in {generation_time:,.2f} seconds.")
+    print(f"Finished generating text in {generation_time:,.2f} seconds.")
     if not isinstance(generated_text, str):
         raise RuntimeError(f"The model {model_name} did not generate any text.")
     if len(generated_text) == 0: