Update app.py

app.py

@@ -6,10 +6,13 @@ client = InferenceClient(
 )
 
 def character_prompt(dict, max_new_tokens):
-    system_prompt = f"ensure responses are no longer than {max_new_tokens} tokens."
-    for key, value in dict.items():
-        system_prompt += f"{key}: {value}."
-    return f"answer me like the main character of a book with this description {system_prompt}"
+    system_prompt = f'<SYSTEM> <the person whose name is {dict["name"]} and your description {dict["description"]}.'
+    system_prompt += f'users name is {dict["user_name"]}.'
+    system_prompt += f'make sure that the greeting is only at the first request.'
+    system_prompt += f'add emojis to enhance the emotions of your message.'
+    system_prompt += f'ensure responses are no longer than {max_new_tokens} tokens.>'
+
+    return system_prompt
         
 def generate(prompt, history, name, description, user_name, max_new_tokens):
     generate_kwargs = dict(
@@ -19,7 +22,7 @@ def generate(prompt, history, name, description, user_name, max_new_tokens):
         repetition_penalty=1.0,
         do_sample=True,
     )
-    system_setting = character_prompt({"Your name": name, "Your description": description, "how to call me": user_name}, max_new_tokens)
+    system_setting = character_prompt({"name": name, "description": description, "user_name": user_name}, max_new_tokens)
     formatted_prompt = format_prompt(prompt, history, system_setting)
     stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
     output = ""
@@ -29,10 +32,10 @@ def generate(prompt, history, name, description, user_name, max_new_tokens):
     return output
 
 def format_prompt(prompt, history, system_setting):
-    formatted_prompt = "<s>"
+    formatted_prompt = "<history>"
     for user_prompt, bot_response in history:
-        formatted_prompt += f"[INST] {user_prompt} [/INST] {bot_response} </s> "
-    formatted_prompt += f"[INST] {system_setting}, {prompt} [/INST]"
+        formatted_prompt += f"[INST] {user_prompt} [/INST] {bot_response} </history> "
+    formatted_prompt += f"[INST] {system_setting}, <user>{prompt}</user> [/INST]"
     return formatted_prompt
 
 additional_inputs = [