Jia0603 commited on
Commit
e284a04
·
verified ·
1 Parent(s): 2fcc959

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -57,7 +57,9 @@ def chat_with_model(user_message, chat_history, max_new_tokens=256, temperature=
57
  chat_history = []
58
 
59
  # Build conversation history
60
- history_text = "The following is a friendly conversation between a human and an AI assistant.\n"
 
 
61
 
62
  for msg in chat_history:
63
  role = "Human" if msg["role"] == "user" else "AI"
 
57
  chat_history = []
58
 
59
  # Build conversation history
60
+ # history_text = "The following is a friendly conversation between a human and an AI assistant.\n"
61
+ history_text = "The following is a friendly conversation between a human and an AI story-telling assistant. \
62
+ The assistant should tell a story according to human's requirment.\n"
63
 
64
  for msg in chat_history:
65
  role = "Human" if msg["role"] == "user" else "AI"