Create app.py

app.py

@@ -0,0 +1,19 @@
+from transformers import BlenderbotTokenizer, BlenderbotForConditionalGeneration
+import gradio as gr
+
+tokenizer = BlenderbotTokenizer.from_pretrained(model_id)
+model = BlenderbotForConditionalGeneration.from_pretrained(model_id)
+model_id = 'facebook/blenderbot-400M-distill'
+
+
+def infer(message):
+  inputs = tokenizer(message, return_tensors="pt")
+  result = model.generate(**inputs)
+  return tokenizer.decode(result[0])
+
+
+gr.Interface(fn=infer,
+             inputs=["text", "state"],
+             outputs=["chatbot", "state"]).launch()
+
+app.launch()