Update app.py

app.py

@@ -14,6 +14,8 @@ INTRO = """**Chat with Yoda, Albert Einstein, Elon Musk or Kanye West!**
 
 👀 **Learn more about Starchat LLM:** [starchat-alpha](https://huggingface.co/blog/starchat-alpha)
 
+👀 **Banner images were created with [stable diffusion web](https://stablediffusionweb.com/).**
+
 ➡️️ **Intended Use**: this demo is intended to be a fun showcase of what one can do with HuggingFace Inference API and recent chat models.
 
 ⚠️ **Limitations**: the model can and will produce factually incorrect information, hallucinating facts and actions. As it has not undergone any advanced tuning/alignment, it can produce problematic outputs, especially if prompted to do so. Finally, this demo is limited to a session length of about 1,000 words.
@@ -32,7 +34,7 @@ INSTRUCTIONS_MAPPING = {
     "Kanye West": "The following is a conversation between rapper Kanye West, and a human user, called User. In the following interactions, User and Kanye West will converse in natural language, and Kanye West will answer User's questions. Kanye West is self-centered, arrogant, a self-proclaimed genius and a great musician. Kanye West interrupted an award ceremony for Taylor Swift and ran for president of the united states. The conversation begins.\n",
 }
 RETRY_COMMAND = "/retry"
-STOP_STR = f"\n{USER_NAME}"
+STOP_SEQ = [f"\n{USER_NAME}", "<|end|>"]
 
 def run_model(prompt, model, temperature, top_p):
     api_url = f"https://api-inference.huggingface.co/models/{model}"
@@ -131,10 +133,10 @@ def chat():
 
     gr.Examples(
         [
+            ["Hi Albert! Why did the apple fall from the tree?"],
             ["Hi Yoda! How do I learn the force?"],
             ["Hi Elon! Give me an idea for a new startup."],
             ["Hi Kanye! What will be the theme of your next album?"],
-            ["Hi Albert! Why did the apple fall from the tree?"],
         ],
         inputs=inputs,
         label="Click on any example and press Enter in the input textbox!",
@@ -157,8 +159,11 @@ def chat():
             temperature=temperature,
             top_p=top_p,
         )
-        model_output = model_output[len(prompt):].split(STOP_STR)[0]
+        for stop in STOP_SEQ:
+            model_output = model_output[len(prompt):].split(stop)[0]
         chat_history = chat_history + [[message, model_output]]
+        print(f"User: {message}")
+        print(f"{bot_name}: {model_output}")
         yield chat_history
         return