Update app.py

app.py

@@ -30,8 +30,8 @@ for model_name, model_endpoint in API_ENDPOINTS.items():
 def format(instruction, history, input, preoutput):
     sy_l, sy_r = SPECIAL_SYMBOLS[0], SPECIAL_SYMBOLS[1]
     formatted_history = "".join(f"{sy_l}{message[0]}{sy_r}\n{sy_l}{message[1]}{sy_r}\n" for message in history)
-    formatted_input = f"{sy_l}System: {instruction}{sy_r}\n{formatted_history}{sy_l}{input}{sy_r}\n"
-    return formatted_input, f"{formatted_input}{sy_l}{preoutput}"
+    formatted_input = f"{sy_l}System: {instruction}{sy_r}\n{formatted_history}{sy_l}{input}{sy_r}\n{sy_l}"
+    return f"{formatted_input}{preoutput}", formatted_input
     
 def predict(instruction, history, input, preoutput, access_key, model, temperature, top_p, top_k, rep_p, max_tokens, stop_seqs, seed):