Staticaliza commited on
Commit
e2ad91e
1 Parent(s): 514b2f9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -33,9 +33,10 @@ for model_name, model_endpoint in API_ENDPOINTS.items():
33
 
34
  def format(instruction, history, input, wrap):
35
  sy_l, sy_r = SPECIAL_SYMBOLS[0], SPECIAL_SYMBOLS[1]
 
36
  formatted_history = "".join(f"{sy_l}{message[0]}{sy_r}\n{sy_l}{message[1]}{sy_r}\n" for message in history)
37
  formatted_input = f"{sy_l}INSTRUCTIONS: {instruction}{sy_r}\n{formatted_history}{sy_l}{input}{sy_r}\n{sy_l}"
38
- return f"{formatted_input}{wrap % ("")}", formatted_input
39
 
40
  def predict(access_key, instruction, history, input, wrap, model, temperature, top_p, top_k, rep_p, max_tokens, stop_seqs, seed):
41
 
 
33
 
34
  def format(instruction, history, input, wrap):
35
  sy_l, sy_r = SPECIAL_SYMBOLS[0], SPECIAL_SYMBOLS[1]
36
+ wrapped_input = wrap % ("")
37
  formatted_history = "".join(f"{sy_l}{message[0]}{sy_r}\n{sy_l}{message[1]}{sy_r}\n" for message in history)
38
  formatted_input = f"{sy_l}INSTRUCTIONS: {instruction}{sy_r}\n{formatted_history}{sy_l}{input}{sy_r}\n{sy_l}"
39
+ return f"{formatted_input}{wrapped_input}", formatted_input
40
 
41
  def predict(access_key, instruction, history, input, wrap, model, temperature, top_p, top_k, rep_p, max_tokens, stop_seqs, seed):
42