:lipstick: update UI

app.py

@@ -46,7 +46,7 @@ cwd = Path.cwd()
 my_cwd = str(cwd.resolve())  # string so it can be passed to os.path() objects
 
 
-def chat(trivia_query, temperature=0.7, top_p=0.95, top_k=50):
+def chat(prompt_message, temperature=0.7, top_p=0.95, top_k=50):
     """
     chat - helper function that makes the whole gradio thing work.
 
@@ -58,13 +58,13 @@ def chat(trivia_query, temperature=0.7, top_p=0.95, top_k=50):
     """
     history = []
     response = ask_gpt(
-        message=trivia_query,
+        message=prompt_message,
         chat_pipe=my_chatbot,
         top_p=top_p,
         top_k=top_k,
         temperature=temperature,
     )
-    history = [trivia_query, response]
+    history = [prompt_message, response]
     html = ""
     for item in history:
         html += f"<b>{item}</b> <br>"
@@ -202,7 +202,7 @@ if __name__ == "__main__":
                 minimum=0.0, maximum=1.0, step=0.01, default=0.6, label="temperature"
             ),
             Slider(minimum=0.0, maximum=1.0, step=0.01, default=0.95, label="top_p"),
-            Slider(minimum=0, maximum=250, step=1, default=50, label="top_k"),
+            Slider(minimum=0, maximum=250, step=5, default=50, label="top_k"),
         ],
         outputs="html",
         examples_per_page=8,
@@ -212,10 +212,10 @@ if __name__ == "__main__":
             ["Hi, my name is……", 0.6, 0.95, 50],
             ["Happy birthday!", 0.9, 0.95, 50],
             ["I have a question, can you help me?", 0.6, 0.95, 50],
-            ["Do you know a joke?", 0.6, 0.85, 50],
-            ["Will you marry me?", 0.6, 0.95, 138],
+            ["Do you know a joke?", 0.8, 0.85, 50],
+            ["Will you marry me?", 0.9, 0.95, 138],
             ["Are you single?", 0.6, 0.95, 138],
-            ["Do you like people?", 0.6, 0.95, 50],
+            ["Do you like people?", 0.7, 0.95, 25],
             ["Are you part of the Matrix?", 0.6, 0.95, 50],
             ["Do you have a hobby?", 0.6, 0.95, 50],
             ["You’re clever", 0.6, 0.95, 50],
@@ -228,12 +228,12 @@ if __name__ == "__main__":
         title=f"GPT Chatbot Demo: {default_model} Model",
         description=f"A Demo of a Chatbot trained for conversation with humans. Size XL= 1.5B parameters.\n\n"
         "**Important Notes & About:**\n\n"
-        "You can find a link to the model card [here](https://huggingface.co/ethzanalytics/ai-msgbot-gpt2-XL-dialogue)\n\n"
-        "1. the model can take up to 60 seconds to respond sometimes, patience is a virtue.\n"
-        "2. the model started from a pretrained checkpoint, and was trained on several different datasets. Anything it says should be fact-checked before being regarded as a true statement.\n"
-        "3. Some params are still being tweaked (in the future, will be inputs) any feedback is welcome :)\n",
+        "You can find a link to the model card **[here](https://huggingface.co/ethzanalytics/ai-msgbot-gpt2-XL-dialogue)**\n\n"
+        "1. responses can take up to 60 seconds to respond sometimes, patience is a virtue.\n"
+        "2. the model was trained on several different datasets.  fact-check responses instead of regarding as a true statement.\n"
+        "3. Try adjusting the **[generation parameters](https://huggingface.co/blog/how-to-generate)** to get a better understanding of how they work!\n",
         css="""
-            .chatbox {display:flex;flex-direction:column}
+            .chatbox {display:flex;flex-direction:row}
             .user_msg, .resp_msg {padding:4px;margin-bottom:4px;border-radius:4px;width:80%}
             .user_msg {background-color:cornflowerblue;color:white;align-self:start}
             .resp_msg {background-color:lightgray;align-self:self-end}