Update app.py

app.py

@@ -33,6 +33,13 @@ tokenizer = transformers.AutoTokenizer.from_pretrained(
     #use_auth_token=HF_AUTH
 )
 
+DESCRIPTION = """
+# StableBeluga2 70B Chat
+This is a streaming Chat Interface implementation of [StableBeluga2](https://huggingface.co/stabilityai/StableBeluga2) 
+You can modify the system prompt, which can be quite fun. For example, you can try something like "You are a mean AI. Phrase all replies as insults" for a good laugh.
+
+Sometimes the model doesn't appropriately hit its stop token. Feel free to hit "stop" and "retry" if this happens to you. Or PR a fix to stop the stream if the tokens for User: get hit or something.
+"""
 
 def prompt_build(system_prompt, user_inp, hist):
     prompt = f"""### System:\n{system_prompt}\n\n"""
@@ -70,6 +77,7 @@ def chat(user_input, history, system_prompt):
 
 
 with gr.Blocks() as demo:
+    gr.Markdown(DESCRIPTION)
     system_prompt = gr.Textbox("You are helpful AI.", label="System Prompt")
     chatbot = gr.ChatInterface(fn=chat, additional_inputs=[system_prompt])