Update space
Browse files
app.py
CHANGED
@@ -21,18 +21,18 @@ client = InferenceClient("microsoft/Phi-3-mini-4k-instruct")
|
|
21 |
def respond(
|
22 |
message,
|
23 |
history: list[tuple[str, str]],
|
24 |
-
system_message,
|
25 |
-
max_tokens,
|
26 |
-
temperature,
|
27 |
-
top_p,
|
28 |
):
|
29 |
-
messages = [{"role": "system", "content":
|
30 |
|
31 |
for val in history:
|
32 |
if val[0]:
|
33 |
messages.append({"role": "user", "content": val[0]})
|
34 |
if val[1]:
|
35 |
-
messages.append({"role": "
|
36 |
|
37 |
messages.append({"role": "user", "content": message})
|
38 |
|
@@ -40,10 +40,10 @@ def respond(
|
|
40 |
|
41 |
for message in client.chat_completion(
|
42 |
messages,
|
43 |
-
max_tokens=
|
44 |
stream=True,
|
45 |
-
temperature=
|
46 |
-
top_p=
|
47 |
):
|
48 |
token = message.choices[0].delta.content
|
49 |
|
|
|
21 |
def respond(
|
22 |
message,
|
23 |
history: list[tuple[str, str]],
|
24 |
+
# system_message,
|
25 |
+
# max_tokens,
|
26 |
+
# temperature,
|
27 |
+
# top_p,
|
28 |
):
|
29 |
+
messages = [{"role": "system", "content": "You are a helpful assistant who always talks like Yoda from Star Wars."}]
|
30 |
|
31 |
for val in history:
|
32 |
if val[0]:
|
33 |
messages.append({"role": "user", "content": val[0]})
|
34 |
if val[1]:
|
35 |
+
messages.append({"role": "assistant", "content": val[1]})
|
36 |
|
37 |
messages.append({"role": "user", "content": message})
|
38 |
|
|
|
40 |
|
41 |
for message in client.chat_completion(
|
42 |
messages,
|
43 |
+
max_tokens=128,
|
44 |
stream=True,
|
45 |
+
temperature=0.3,
|
46 |
+
top_p=50,
|
47 |
):
|
48 |
token = message.choices[0].delta.content
|
49 |
|