Haris-Subrata commited on
Commit
2bc1ce7
·
verified ·
1 Parent(s): 021c767

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -283,13 +283,12 @@ with gr.Blocks() as demo:
283
  def user_turn(user_message: str, chat_history: list) -> tuple[str, list]:
284
  if not user_message.strip():
285
  return "", chat_history
286
- return "", chat_history + [(user_message.strip(), None)]
287
 
288
  def bot_turn(chat_history: list, temperature_value: float, top_p_value: float) -> list:
289
- user_message = chat_history[-1][0]
290
  assistant_message = respond(user_message, chat_history[:-1], temperature_value, top_p_value)
291
- chat_history[-1] = (user_message, assistant_message)
292
- return chat_history
293
 
294
  submit.click(user_turn, [message, chatbot], [message, chatbot], queue=False).then(
295
  bot_turn,
 
283
  def user_turn(user_message: str, chat_history: list) -> tuple[str, list]:
284
  if not user_message.strip():
285
  return "", chat_history
286
+ return "", chat_history + [{"role": "user", "content": user_message.strip()}]
287
 
288
  def bot_turn(chat_history: list, temperature_value: float, top_p_value: float) -> list:
289
+ user_message = chat_history[-1]["content"]
290
  assistant_message = respond(user_message, chat_history[:-1], temperature_value, top_p_value)
291
+ return chat_history + [{"role": "assistant", "content": assistant_message}]
 
292
 
293
  submit.click(user_turn, [message, chatbot], [message, chatbot], queue=False).then(
294
  bot_turn,