Update app.py

app.py

@@ -13,8 +13,8 @@ def format_prompt(message, history):
     prompt_prefix = "Please correct the grammar in the following sentence:"
     prompt_template = "[INST] " + prompt_prefix + " {} [/INST]"
     
-    print("History Type: {}".format(type(history)))
-    #history.append("It is my friends house in England.", "It is my friend's house in England.")
+    history.append("It is my friends house in England.", "It is my friend's house in England.")
+    history.append("Every girl must bring their books to school.", "Every girl must bring her books to school.")
 
     # Iterates through every past user input and response to be added to the prompt
     for user_prompt, bot_response in history:
@@ -28,14 +28,14 @@ def format_prompt(message, history):
     return prompt
 
 def generate(prompt, history, system_prompt, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0):
-    print("System Prompt: '{}'".format(system_prompt))
+    print("\nSystem Prompt: '{}'".format(system_prompt))
     temperature = float(temperature)
     if temperature < 1e-2:
         temperature = 1e-2
     top_p = float(top_p)
 
     generate_kwargs = dict(temperature=temperature, max_new_tokens=max_new_tokens, top_p=top_p, repetition_penalty=repetition_penalty, do_sample=True, seed=42,)
-
+    
     formatted_prompt = format_prompt(f"{system_prompt}, {prompt}", history)
     stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
     output = ""
@@ -46,6 +46,7 @@ def generate(prompt, history, system_prompt, temperature=0.9, max_new_tokens=256
     return output
 
 
+
 additional_inputs=[
     gr.Textbox( label="System Prompt", value="Correct the following sentence to make it grammatically accurate while maintaining the original meaning. Output only the corrected sentence." , max_lines=1, interactive=True, ),
     gr.Slider( label="Temperature", value=0.9, minimum=0.0, maximum=1.0, step=0.05, interactive=True, info="Higher values produce more diverse outputs", ),