Update app.py

app.py

@@ -23,18 +23,21 @@ client = InferenceClient(
 def format_prompt(message, history):
     prompt = "<s>"
 
-    # Prepend "Please correct the grammar in the following sentence: " to each user input
-    grammar_correction_prefix = "Please correct the grammar in the following sentence: "
-
+    # String to add before every prompt
+    prompt_prefix = "Please correct the grammar in the following sentence: "
+    
     # Iterates through every past user input and response to be added to the prompt
     for user_prompt, bot_response in history:
-        corrected_user_prompt = grammar_correction_prefix + user_prompt
-        prompt += f"[INST] {corrected_user_prompt} [/INST]"
+        corrected_prompt = prompt_prefix + user_prompt
+        
+        prompt += f"[INST] {corrected_prompt} [/INST]"
         prompt += f" {bot_response}</s> "
+        #print(f"HISTORIC PROMPT: \n\t[INST] {corrected_prompt} [/INST] {bot_response}</s> ")
 
     # Also prepend the prefix to the current message
     corrected_message = grammar_correction_prefix + message
     prompt += f"[INST] {corrected_message} [/INST]"
+    print("\nPROMPT: \n\t" + prompt)
 
     return prompt