Update app.py

app.py

@@ -9,7 +9,7 @@ from huggingface_hub import Repository, InferenceClient
 HF_TOKEN = os.environ.get("HF_TOKEN", None)
 API_URL = "https://api-inference.huggingface.co/models/meta-llama/Llama-2-70b-chat-hf"
 API_URL_2 = "https://api-inference.huggingface.co/models/codellama/CodeLlama-34b-Instruct-hf"
-BOT_NAME = "LLAMA"
+BOT_NAME = "Assistant"
 
 STOP_SEQUENCES = ["\nUser:", " User:", "###", "</s>"]
 
@@ -35,13 +35,13 @@ def format_prompt(message, history, system_prompt):
   prompt = ""
   if system_prompt:
     prompt += f"System: {system_prompt}\n"
-#  for user_prompt, bot_response in history:
-#    prompt += f"User: {user_prompt}\n"
-#    prompt += f"LLAMA: {bot_response}\n" # Response already contains "Falcon: "
+  for user_prompt, bot_response in history:
+    prompt += f"User: {user_prompt}\n"
+    prompt += f"{BOT_NAME}: {bot_response}\n"
   prompt += f"""User: {message}\n{BOT_NAME}:"""
   return prompt
 
-seed = 42442
+seed = 42
 
 def generate(
     prompt, history, system_prompt="", temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0,