Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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"
|