AkashKhatri commited on
Commit
b775ff3
·
verified ·
1 Parent(s): c39ccb2

Delete chatbot.py

Browse files
Files changed (1) hide show
  1. chatbot.py +0 -12
chatbot.py DELETED
@@ -1,12 +0,0 @@
1
- # chatbot.py
2
- from transformers import BlenderbotTokenizer, BlenderbotForConditionalGeneration
3
-
4
- tokenizer = BlenderbotTokenizer.from_pretrained("facebook/blenderbot-400M-distill")
5
- model = BlenderbotForConditionalGeneration.from_pretrained("facebook/blenderbot-400M-distill")
6
-
7
- def generate_response(user_input, chat_history=[]):
8
- inputs = tokenizer([user_input], return_tensors='pt')
9
- chat_history_ids = model.generate(**inputs)
10
- bot_response = tokenizer.decode(chat_history_ids[0], skip_special_tokens=True)
11
- chat_history.append((user_input, bot_response))
12
- return bot_response, chat_history