mikepastor11 commited on
Commit
b44a90c
1 Parent(s): 84fdd41

example code added

Browse files
Files changed (1) hide show
  1. app.py +14 -4
app.py CHANGED
@@ -1,8 +1,18 @@
 
1
  import gradio as gr
2
 
3
- def greet(name):
4
- return "Hello there " + name + "!!"
5
 
6
- iface = gr.Interface(fn=greet, inputs="text", outputs="text")
 
 
 
 
 
 
 
 
 
 
 
7
 
8
- iface.launch()
 
1
+ from transformers import pipeline, Conversation
2
  import gradio as gr
3
 
4
+ chatbot = pipeline(model="facebook/blenderbot-400M-distill")
 
5
 
6
+ message_list = []
7
+ response_list = []
8
+
9
+ def vanilla_chatbot(message, history):
10
+ conversation = Conversation(text=message, past_user_inputs=message_list, generated_responses=response_list)
11
+ conversation = chatbot(conversation)
12
+
13
+ return conversation.generated_responses[-1]
14
+
15
+ demo_chatbot = gr.ChatInterface(vanilla_chatbot, title="Vanilla Chatbot", description="Enter text to start chatting.")
16
+
17
+ demo_chatbot.launch()
18