smjain commited on
Commit
b4f0646
·
1 Parent(s): 2ca7a33

Create new file

Browse files
Files changed (1) hide show
  1. app.py +22 -0
app.py ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import BlenderbotTokenizer, BlenderbotForConditionalGeneration
2
+ import gradio as grad
3
+
4
+ chat_tkn = BlenderbotTokenizer.from_pretrained("facebook/blenderbot-400M-distill")
5
+ mdl = BlenderbotForConditionalGeneration.from_pretrained("facebook/blenderbot-400M-distill")
6
+
7
+ def chat():
8
+ for step in range(5):
9
+
10
+ tkn_ids = chat_tkn(input(">> User:") + chat_tkn.eos_token, return_tensors='pt')
11
+
12
+ # bot responds
13
+ chat_ids = mdl.generate(**tkn_ids)
14
+
15
+ # print bot response
16
+ response= "Alicia: {}".format(chat_tkn.decode(chat_ids[0], skip_special_tokens=True)))
17
+
18
+ return response
19
+
20
+ out=grad.Textbox(lines=20, label="dialog", placeholder="start conversation")
21
+ grad.Interface(chat, outputs=out).launch()
22
+