JoeStrout commited on
Commit
b59c36e
Β·
verified Β·
1 Parent(s): f1b20f5

Upload app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -197,9 +197,9 @@ def chat(message: str, history: list) -> str:
197
  system_prompt = build_system_prompt(results)
198
 
199
  messages = [{"role": "system", "content": system_prompt}]
200
- for user_msg, assistant_msg in history:
201
- messages.append({"role": "user", "content": user_msg})
202
- messages.append({"role": "assistant", "content": assistant_msg})
203
  messages.append({"role": "user", "content": message})
204
 
205
  text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
 
197
  system_prompt = build_system_prompt(results)
198
 
199
  messages = [{"role": "system", "content": system_prompt}]
200
+ # Gradio 5: history is list of {"role": ..., "content": ...} dicts
201
+ for entry in history:
202
+ messages.append({"role": entry["role"], "content": entry["content"]})
203
  messages.append({"role": "user", "content": message})
204
 
205
  text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)