Create chatbot.py

chatbot.py

@@ -0,0 +1,12 @@
+# 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