test

app.py

@@ -5,14 +5,17 @@ client = InferenceClient(
     "mistralai/Mistral-7B-Instruct-v0.3"
 )
 
+# Your system prompt
+SYSTEM_PROMPT = "You are an intelligent assistant. Please provide useful and concise answers.\n"
 
 def format_prompt(message, history):
-  prompt = "<s>"
-  for user_prompt, bot_response in history:
-    prompt += f"[INST] {user_prompt} [/INST]"
-    prompt += f" {bot_response}</s> "
-  prompt += f"[INST] {message} [/INST]"
-  return prompt
+    prompt = "<s>"
+    prompt += f"[INST] SYSTEM: {SYSTEM_PROMPT} [/INST]"  # Add the system prompt here
+    for user_prompt, bot_response in history:
+        prompt += f"[INST] {user_prompt} [/INST]"
+        prompt += f" {bot_response}</s> "
+    prompt += f"[INST] {message} [/INST]"
+    return prompt
 
 def generate(
     prompt, history, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0,
@@ -41,7 +44,6 @@ def generate(
         yield output
     return output
 
-
 additional_inputs=[
     gr.Slider(
         label="Temperature",
@@ -81,10 +83,9 @@ additional_inputs=[
     )
 ]
 
-
 gr.ChatInterface(
     fn=generate,
     chatbot=gr.Chatbot(show_label=False, show_share_button=False, show_copy_button=True, likeable=True, layout="panel"),
     additional_inputs=additional_inputs,
-    title="""Mistral 7B v0.3"""
+    title="""AI Paragraph Generator"""
 ).launch(show_api=False)