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)
|