Update app.py

app.py

@@ -9,14 +9,14 @@ client = InferenceClient(
 def format_prompt(message, history):
     prompt = "<s>"
     for user_prompt, bot_response in history:
+        print("USER PROMPT: '{}' \nBOT RESPONSE: '{}'".format(user_prompt, bot_response))
         prompt += f"[INST] {user_prompt} [/INST]"
         prompt += f" {bot_response}</s> "
+    print("MSG: '{}'".format(message))
     prompt += f"[INST] {message} [/INST]"
     return prompt
 
 def generate(prompt, history, system_prompt, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0):
-    dashes = "-"*100 + "\n"
-    print(dashes + "\n\nHELLO THIS IS COLE\n\n\n" + dashes)
     temperature = float(temperature)
     if temperature < 1e-2:
         temperature = 1e-2