Improver user messages

hanlde_form_submit.py

@@ -14,13 +14,34 @@ def create_pipeline(model_name: str, group_size) -> GroupedSamplingPipeLine:
     :param group_size: The size of the groups to use.
     :return: A pipeline with the given model name and group size.
     """
-    return GroupedSamplingPipeLine(
+    print(f"Starts downloading model: {model_name} from the internet.")
+    pipeline =  GroupedSamplingPipeLine(
         model_name=model_name,
         group_size=group_size,
         end_of_sentence_stop=True,
         temp=0.5,
         top_p=0.6,
     )
+    print(f"Finished downloading model: {model_name} from the internet.")
+    return pipeline
+
+
+def generate_text(
+        pipeline: GroupedSamplingPipeLine,
+        prompt: str,
+        output_length: int,
+) -> str:
+    """
+    Generates text using the given pipeline.
+    :param pipeline: The pipeline to use.
+    :param prompt: The prompt to use.
+    :param output_length: The size of the groups to use.
+    :return: The generated text.
+    """
+    return pipeline(
+        prompt_s=prompt,
+        max_new_tokens=output_length,
+    )["generated_text"]
 
 
 @st.cache
@@ -40,7 +61,8 @@ def on_form_submit(model_name: str, output_length: int, prompt: str) -> str:
         model_name=model_name,
         group_size=output_length,
     )
-    return pipeline(
-        prompt_s=prompt,
-        max_new_tokens=output_length,
-    )["generated_text"]
+    return generate_text(
+        pipeline=pipeline,
+        prompt=prompt,
+        output_length=output_length,
+    )