Spaces:
Sleeping
Sleeping
Staticaliza
commited on
Commit
•
b155bbd
1
Parent(s):
b3f87df
Update app.py
Browse files
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.
|
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}
|
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 |
|