research14 commited on
Commit
68440cc
·
1 Parent(s): 02f452d

undo last change

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -64,7 +64,7 @@ def vicuna_respond(tab_name, message, chat_history):
64
  bot_message = vicuna_tokenizer.decode(output_ids[0], skip_special_tokens=True)
65
  print(bot_message)
66
 
67
- chat_history.append((bot_message))
68
  time.sleep(2)
69
  return tab_name, "", chat_history
70
 
@@ -77,7 +77,7 @@ def llama_respond(tab_name, message, chat_history):
77
  bot_message = llama_tokenizer.decode(output_ids[0], skip_special_tokens=True)
78
  print(bot_message)
79
 
80
- chat_history.append((bot_message))
81
  time.sleep(2)
82
  return tab_name, "", chat_history
83
 
 
64
  bot_message = vicuna_tokenizer.decode(output_ids[0], skip_special_tokens=True)
65
  print(bot_message)
66
 
67
+ chat_history.append((formatted_prompt, bot_message))
68
  time.sleep(2)
69
  return tab_name, "", chat_history
70
 
 
77
  bot_message = llama_tokenizer.decode(output_ids[0], skip_special_tokens=True)
78
  print(bot_message)
79
 
80
+ chat_history.append((formatted_prompt, bot_message))
81
  time.sleep(2)
82
  return tab_name, "", chat_history
83