Staticaliza commited on
Commit
b155bbd
1 Parent(s): b3f87df

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -13,7 +13,7 @@ SPECIAL_SYMBOLS = ["‹", "›"]
13
 
14
  DEFAULT_INPUT = f"You: Hi!"
15
  DEFAULT_PREOUTPUT = f"AI: "
16
- DEFAULT_INSTRUCTION = "You are an helpful chatbot. You must respond in this format: \"‹NAME: MESSAGE›\""
17
 
18
  API_ENDPOINTS = {
19
  "Falcon": "tiiuae/falcon-180B-chat",
@@ -30,7 +30,7 @@ for model_name, model_endpoint in API_ENDPOINTS.items():
30
  def format(instruction = DEFAULT_INSTRUCTION, history = [], input = "", preoutput = ""):
31
  sy_l, sy_r = SPECIAL_SYMBOLS[0], SPECIAL_SYMBOLS[1]
32
  formatted_history = '\n'.join(f"{sy_l}{message[0]}{sy_r}\n{sy_l}{message[1]}{sy_r}" for message in history)
33
- formatted_input = f"{sy_l}System: {instruction}{sy_r}\n{formatted_history}\n{sy_l}{input}{sy_r}\n{sy_l}{preoutput}"
34
  print(formatted_input)
35
  return formatted_input
36
 
 
13
 
14
  DEFAULT_INPUT = f"You: Hi!"
15
  DEFAULT_PREOUTPUT = f"AI: "
16
+ DEFAULT_INSTRUCTION = "You are an helpful chatbot."
17
 
18
  API_ENDPOINTS = {
19
  "Falcon": "tiiuae/falcon-180B-chat",
 
30
  def format(instruction = DEFAULT_INSTRUCTION, history = [], input = "", preoutput = ""):
31
  sy_l, sy_r = SPECIAL_SYMBOLS[0], SPECIAL_SYMBOLS[1]
32
  formatted_history = '\n'.join(f"{sy_l}{message[0]}{sy_r}\n{sy_l}{message[1]}{sy_r}" for message in history)
33
+ formatted_input = f"{sy_l}System: {instruction}{sy_r}{formatted_history}\n{sy_l}{input}{sy_r}\n{sy_l}{preoutput}"
34
  print(formatted_input)
35
  return formatted_input
36