Spaces:
Sleeping
Sleeping
new message format2
Browse files
app.py
CHANGED
@@ -25,7 +25,7 @@ def predict(message, history):
|
|
25 |
# messages = "".join(["".join(["<|start_header_id|>user<|end_header_id|>\n"+item[0],
|
26 |
# "<|eot_id|><|start_header_id|>assistant<|end_header_id|>\n"+item[1]])
|
27 |
# for item in history_transformer_format])
|
28 |
-
messages = [{"
|
29 |
print(messages)
|
30 |
|
31 |
# model_inputs = tokenizer([messages], return_tensors="pt") # .to("cuda")
|
|
|
25 |
# messages = "".join(["".join(["<|start_header_id|>user<|end_header_id|>\n"+item[0],
|
26 |
# "<|eot_id|><|start_header_id|>assistant<|end_header_id|>\n"+item[1]])
|
27 |
# for item in history_transformer_format])
|
28 |
+
messages = [{"user": item[0], "content": item[1]} for item in history_transformer_format]
|
29 |
print(messages)
|
30 |
|
31 |
# model_inputs = tokenizer([messages], return_tensors="pt") # .to("cuda")
|