Create app.py

app.py

@@ -0,0 +1,36 @@
+from transformers import BlenderbotTokenizer, BlenderbotForConditionalGeneration, pipeline
+import gradio as gr
+
+mname = "facebook/blenderbot-400M-distill"
+
+model = BlenderbotForConditionalGeneration.from_pretrained(mname)
+
+tokenizer = BlenderbotTokenizer.from_pretrained(mname)
+
+to_english = pipeline('translation', model='Helsinki-NLP/opus-mt-es-en')
+
+to_spanish = pipeline('translation', model='Helsinki-NLP/opus-mt-en-es')
+
+def chat(msg, history):
+
+    history = history or []
+    mensaje = msg
+    MESSAGE = to_english(mensaje)
+    inputs = tokenizer([MESSAGE[0]['translation_text']], return_tensors="pt")
+    reply_ids = model.generate(**inputs)
+    respuesta = to_spanish(tokenizer.batch_decode(reply_ids))
+    history.append((msg,respuesta[0]['translation_text']))
+    
+    return history, history
+
+chatbot = gr.Chatbot().style(color_map=("red", "blue"))
+demo = gr.Interface(
+    chat,
+    ["text", "state"],
+    [chatbot, "state"],
+    allow_flagging="never",
+)
+
+
+if __name__ == "__main__":
+    demo.launch()