ryderprogram commited on
Commit
d9c8b41
·
1 Parent(s): ce0bcb9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -1,3 +1,7 @@
1
  import gradio as gr
 
2
 
3
- gr.Interface.load("models/facebook/blenderbot-400M-distill").launch()
 
 
 
 
1
  import gradio as gr
2
+ from transformers import pipeline, Conversation
3
 
4
+ pipeline = pipeline(task="conversational", model="facebook/blenderbot-400M-distill")
5
+ con1 = Conversation("Going to the movies tonight - any suggestions?")
6
+ con2 = Conversation("What's the last book you have read?")
7
+ converse([con1, con2])