nanova commited on
Commit
e6d4457
·
1 Parent(s): 2d0b3a0
Files changed (1) hide show
  1. app.py +8 -10
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[tuple[str, str]],
29
  system_message,
30
  max_tokens,
31
  temperature,
@@ -33,12 +33,10 @@ def respond(
33
  ):
34
  messages = [{"role": "system", "content": system_message}]
35
 
36
- for val in history:
37
- if val[0]:
38
- messages.append({"role": "user", "content": val[0]})
39
- if val[1]:
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)