Update app.py

app.py

@@ -34,6 +34,15 @@ tokenizer = transformers.AutoTokenizer.from_pretrained(
 )
 
 
+DESCRIPTION = """
+# Stable Beluga 7B Chat
+This is a streaming Chat Interface implementation of [StableBeluga-7B](https://huggingface.co/stabilityai/StableBeluga-7B) 
+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 +79,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])