Update app.py

app.py

@@ -36,12 +36,12 @@ 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.
+This is a streaming Chat Interface implementation of [StableBeluga-7B](https://huggingface.co/stabilityai/StableBeluga-7B). We'll use it to deploy a Discord bot that you can add to your server!
 
 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.
 """
 
+system_prompt = "You are helpful AI."
 
 def prompt_build(system_prompt, user_inp, hist):
     prompt = f"""### System:\n{system_prompt}\n\n"""
@@ -52,7 +52,7 @@ def prompt_build(system_prompt, user_inp, hist):
     prompt += f"""### User:\n{user_inp}\n\n### Assistant:"""
     return prompt
 
-def chat(user_input, history, system_prompt):
+def chat(user_input, history):
 
     prompt = prompt_build(system_prompt, user_input, history)
     model_inputs = tokenizer([prompt], return_tensors="pt").to("cuda")
@@ -80,7 +80,6 @@ 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])
+    chatbot = gr.ChatInterface(fn=chat)
 
 demo.queue().launch()