Spaces:
Sleeping
Sleeping
| # chatbot.py | |
| from transformers import BlenderbotTokenizer, BlenderbotForConditionalGeneration | |
| tokenizer = BlenderbotTokenizer.from_pretrained("facebook/blenderbot-400M-distill") | |
| model = BlenderbotForConditionalGeneration.from_pretrained("facebook/blenderbot-400M-distill") | |
| def generate_response(user_input, chat_history=[]): | |
| inputs = tokenizer([user_input], return_tensors='pt') | |
| chat_history_ids = model.generate(**inputs) | |
| bot_response = tokenizer.decode(chat_history_ids[0], skip_special_tokens=True) | |
| chat_history.append({"role": "assistant", "content": bot_response}) | |
| return bot_response, chat_history | |