Update output with project name

app.py

@@ -322,9 +322,15 @@ def text_to_text_generation(verbose, prompt):
         original, new = generate_batch(prompt)[0]
         prompt, response = cleanup(original, new)
         if verbose:
-            return f"{modes}\n\n" f"# Prompt given to the model:\n" f"{str_to_md(prompt)}\n" f"# Model's answer:\n" f"{str_to_md(response)}\n"
+            return (
+                f"# 📚KubeWizard📚\n"
+                f"{modes}\n\n" f"# Prompt given to the model:\n" f"{str_to_md(prompt)}\n" f"# Model's answer:\n" f"{str_to_md(response)}\n"
+            )
         else:
-            return f"{modes}\n\n" f"# Answer:\n" f"{str_to_md(response)}"
+            return (
+                f"# 📚KubeWizard📚\n"
+                f"{modes}\n\n" f"# Answer:\n" f"{str_to_md(response)}"
+            )
 
     if response_num == 0:
         prompt = create_generation_prompt(response_num, prompt, False)
@@ -334,6 +340,7 @@ def text_to_text_generation(verbose, prompt):
         model_response = new[len(original):].strip()
         if verbose:
             return (
+                f"# 📚KubeWizard📚\n"
                 f"{modes}\n\n"
                 f"# Prompt given to the model:\n"
                 f"{str_to_md(prompt)}\n"
@@ -343,7 +350,10 @@ def text_to_text_generation(verbose, prompt):
                 f"```bash\n{str_to_md(response)}\n```\n"
             )
         else:
-            return f"{modes}\n\n" f"# Answer:\n" f"```bash\n{str_to_md(response)}\n```\n"
+            return (
+                f"# 📚KubeWizard📚\n"
+                f"{modes}\n\n" f"# Answer:\n" f"```bash\n{str_to_md(response)}\n```\n"
+            )
 
     res_prompt = create_generation_prompt(response_num, prompt, False)
     print(f'Prompt given to finetuned model:\n{res_prompt}\n')
@@ -362,6 +372,7 @@ def text_to_text_generation(verbose, prompt):
 
     if verbose:
         return (
+            f"# 📚KubeWizard📚\n"
             f"{modes}\n\n"
             f"# Answer with finetuned model\n"
             f"## Prompt given to the model:\n"
@@ -381,6 +392,7 @@ def text_to_text_generation(verbose, prompt):
         )
     else:
         return (
+            f"# 📚KubeWizard📚\n"
             f"{modes}\n\n"
             f"# Answer with finetuned model\n\n {str_to_md(res_normal)}\n"
             f"# Answer with RAG\n\n {str_to_md(res_semantic_search)}\n"