hikewa commited on
Commit
c9ded2b
·
verified ·
1 Parent(s): 8c26b84

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +6 -4
app.py CHANGED
@@ -41,9 +41,12 @@ def respond(message, history):
41
  load_model()
42
 
43
  messages = [{"role": "system", "content": SYSTEM_PROMPT}]
44
- for user_msg, bot_msg in history:
45
- messages.append({"role": "user", "content": user_msg})
46
- messages.append({"role": "assistant", "content": bot_msg})
 
 
 
47
  messages.append({"role": "user", "content": message})
48
 
49
  text = tokenizer.apply_chat_template(
@@ -80,7 +83,6 @@ demo = gr.ChatInterface(
80
  "Is it better to optimize for individual freedom or collective wellbeing?",
81
  "When does pragmatic compromise become unprincipled capitulation?",
82
  ],
83
- type="tuples",
84
  )
85
 
86
  if __name__ == "__main__":
 
41
  load_model()
42
 
43
  messages = [{"role": "system", "content": SYSTEM_PROMPT}]
44
+ for msg in history:
45
+ if isinstance(msg, dict):
46
+ messages.append(msg)
47
+ elif isinstance(msg, (list, tuple)) and len(msg) == 2:
48
+ messages.append({"role": "user", "content": msg[0]})
49
+ messages.append({"role": "assistant", "content": msg[1]})
50
  messages.append({"role": "user", "content": message})
51
 
52
  text = tokenizer.apply_chat_template(
 
83
  "Is it better to optimize for individual freedom or collective wellbeing?",
84
  "When does pragmatic compromise become unprincipled capitulation?",
85
  ],
 
86
  )
87
 
88
  if __name__ == "__main__":