Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -242,8 +242,7 @@ def simple_chat(message: dict, temperature: float = 0.8, max_length: int = 4096,
|
|
| 242 |
else:
|
| 243 |
conversation.append({"role": "user", "content": message['text']})
|
| 244 |
|
| 245 |
-
|
| 246 |
-
input_ids = tokenizer(conversation[-1]['content'], return_tensors="pt").to(model.device)
|
| 247 |
|
| 248 |
generate_kwargs = dict(
|
| 249 |
max_length=max_length,
|
|
|
|
| 242 |
else:
|
| 243 |
conversation.append({"role": "user", "content": message['text']})
|
| 244 |
|
| 245 |
+
input_ids = tokenizer(input_text, return_tensors="pt").to(model.device)
|
|
|
|
| 246 |
|
| 247 |
generate_kwargs = dict(
|
| 248 |
max_length=max_length,
|