Spaces:
Running
Running
update
Browse files
app.py
CHANGED
@@ -25,7 +25,7 @@ For more information on `huggingface_hub` Inference API support, please check th
|
|
25 |
|
26 |
def respond(
|
27 |
message,
|
28 |
-
history: list[
|
29 |
system_message,
|
30 |
max_tokens,
|
31 |
temperature,
|
@@ -33,12 +33,10 @@ def respond(
|
|
33 |
):
|
34 |
messages = [{"role": "system", "content": system_message}]
|
35 |
|
36 |
-
|
37 |
-
|
38 |
-
|
39 |
-
|
40 |
-
messages.append({"role": "assistant", "content": val[1]})
|
41 |
-
|
42 |
messages.append({"role": "user", "content": message})
|
43 |
|
44 |
headers = {
|
@@ -48,7 +46,7 @@ def respond(
|
|
48 |
|
49 |
data = {
|
50 |
"model": "/data/DMind-1-mini",
|
51 |
-
"stream": False,
|
52 |
"messages": messages,
|
53 |
"temperature": temperature,
|
54 |
"top_p": top_p,
|
@@ -87,7 +85,6 @@ demo = gr.ChatInterface(
|
|
87 |
respond,
|
88 |
additional_inputs=[
|
89 |
gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
|
90 |
-
gr.Slider(minimum=1, maximum=32768, value=16384, step=1, label="Max new tokens"),
|
91 |
gr.Slider(minimum=0.1, maximum=4.0, value=0.6, step=0.1, label="Temperature"),
|
92 |
gr.Slider(
|
93 |
minimum=0.1,
|
@@ -97,8 +94,9 @@ demo = gr.ChatInterface(
|
|
97 |
label="Top-p (nucleus sampling)",
|
98 |
),
|
99 |
],
|
|
|
100 |
)
|
101 |
|
102 |
|
103 |
if __name__ == "__main__":
|
104 |
-
demo.launch()
|
|
|
25 |
|
26 |
def respond(
|
27 |
message,
|
28 |
+
history: list[dict], # 修改为新的消息格式
|
29 |
system_message,
|
30 |
max_tokens,
|
31 |
temperature,
|
|
|
33 |
):
|
34 |
messages = [{"role": "system", "content": system_message}]
|
35 |
|
36 |
+
# 添加历史消息
|
37 |
+
messages.extend(history)
|
38 |
+
|
39 |
+
# 添加当前用户消息
|
|
|
|
|
40 |
messages.append({"role": "user", "content": message})
|
41 |
|
42 |
headers = {
|
|
|
46 |
|
47 |
data = {
|
48 |
"model": "/data/DMind-1-mini",
|
49 |
+
"stream": False,
|
50 |
"messages": messages,
|
51 |
"temperature": temperature,
|
52 |
"top_p": top_p,
|
|
|
85 |
respond,
|
86 |
additional_inputs=[
|
87 |
gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
|
|
|
88 |
gr.Slider(minimum=0.1, maximum=4.0, value=0.6, step=0.1, label="Temperature"),
|
89 |
gr.Slider(
|
90 |
minimum=0.1,
|
|
|
94 |
label="Top-p (nucleus sampling)",
|
95 |
),
|
96 |
],
|
97 |
+
type="messages" # 指定使用新的消息格式
|
98 |
)
|
99 |
|
100 |
|
101 |
if __name__ == "__main__":
|
102 |
+
demo.launch(share=True)
|