Staticaliza commited on
Commit
44ae3d6
1 Parent(s): 04fbd0c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ for model_name, model_endpoint in API_ENDPOINTS.items():
26
  def format(instruction = "", history = "", input = "", preinput = ""):
27
  sy_l, sy_r = SPECIAL_SYMBOLS[0], SPECIAL_SYMBOLS[1]
28
  formatted_history = '\n'.join(f"{sy_l}{message}{sy_r}" for message in history)
29
- task_message = f"{instruction}\n{formatted_history}\n{sy_l}{input}{sy_r}\n{preinput}"
30
  return prompt
31
 
32
  def predict(instruction, history, input, preinput, access_key, model, temperature, top_p, top_k, rep_p, max_tokens, stop_seqs, seed):
 
26
  def format(instruction = "", history = "", input = "", preinput = ""):
27
  sy_l, sy_r = SPECIAL_SYMBOLS[0], SPECIAL_SYMBOLS[1]
28
  formatted_history = '\n'.join(f"{sy_l}{message}{sy_r}" for message in history)
29
+ task_message = f"System: {sy_l}{instruction}{sy_r}\n{formatted_history}\n{sy_l}{input}{sy_r}\n{preinput}"
30
  return prompt
31
 
32
  def predict(instruction, history, input, preinput, access_key, model, temperature, top_p, top_k, rep_p, max_tokens, stop_seqs, seed):