Update app.py

app.py

@@ -1,5 +1,27 @@
 import gradio as gr
+from transformers import AutoTokenizer, TFBlenderbotForConditionalGeneration
+import tensorflow as tf
+from transformers import pipeline, Conversation
 
-demo = gr.Interface.load("models/facebook/blenderbot-400M-distill")
+tokenizer = AutoTokenizer.from_pretrained("facebook/blenderbot-400M-distill")
+model = TFBlenderbotForConditionalGeneration.from_pretrained("facebook/blenderbot-400M-distill")
+
+chatbot = pipeline(task="conversational",model=model,tokenizer=tokenizer)
+
+def chatBotPred(message,history):
+  if len(history)>0:
+    past_user_inputs= [chat[0] for chat in history]
+    generated_responses= [chat[1] for chat in history]
+  else:
+    past_user_inputs= None
+    generated_responses= None
+
+  conversation = Conversation(text=message,
+                              past_user_inputs=past_user_inputs,
+                              generated_responses=generated_responses)
+  conversation = chatbot(conversation)
+  return conversation.generated_responses[-1]
+
+demo = gr.ChatInterface(chatBotPred)
 
 demo.launch()