(message)
| 11 | |
| 12 | |
| 13 | def generate_response(message): |
| 14 | inputs = tokenizer(message, return_tensors='pt', truncation=True) |
| 15 | outputs = model.generate( |
| 16 | **inputs, |
| 17 | max_length=150, |
| 18 | pad_token_id=tokenizer.eos_token_id |
| 19 | ) |
| 20 | response = tokenizer.decode(outputs[0], skip_special_tokens=True) |
| 21 | return response |
| 22 | |
| 23 | # Function to handle when the user sends a message |
| 24 |