fix

gemmademo/_chat.py

@@ -96,7 +96,9 @@ class GradioChat:
         with gr.Blocks() as demo:
             with gr.Row():
                 with gr.Column(scale=3):  # Sidebar column
-                    gr.Markdown("## Google Gemma Models")
+                    gr.Markdown(
+                        "## Google Gemma Models: lightweight, state-of-the-art open models from Google"
+                    )
                     task_dropdown = gr.Dropdown(
                         choices=self.task_options,
                         value=self.current_task_name,

gemmademo/_prompts.py

@@ -1,4 +1,11 @@
 class PromptManager:
+    """
+    A manager for generating prompts based on different tasks.
+
+    This class provides methods to format user input into prompts suitable for
+    various tasks such as Question Answering, Text Generation, and Code Completion.
+    It raises a ValueError if an unsupported task is specified.
+    """
     def __init__(self, task):
         self.task = task