Upload folder using huggingface_hub

app.py

@@ -0,0 +1,24 @@
+import gradio as gr
+from transformers import pipeline
+
+model_name = "microsoft/DialoGPT-medium"
+
+def converse(message, chat_history):
+    if chat_history is None:
+        chat_history = []
+    conversational_pipeline = pipeline('conversational', model=model_name, use_auth_token=None)
+    from transformers import Conversation
+    convo = Conversation(message)
+    result = conversational_pipeline(convo)
+    return result.generated_responses[-1], chat_history + [[message, result.generated_responses[-1]]]
+
+with gr.Blocks() as demo:
+    gr.Markdown("# Dialogue Bot")
+    gr.Markdown("Powered by microsoft/DialoGPT-medium")
+    chatbot = gr.Chatbot()
+    msg = gr.Textbox(label="Message", placeholder="Type your message...")
+    btn = gr.Button("Send")
+    btn.click(fn=converse, inputs=[msg, chatbot], outputs=[msg, chatbot])
+
+if __name__ == "__main__":
+    demo.launch()

requirements.txt

@@ -0,0 +1,3 @@
+gradio
+transformers
+torch