initial commit

app.py

@@ -0,0 +1,22 @@
+import gradio as gr
+from transformers import pipeline, Conversation
+
+chatbot = pipeline("conversational", model="facebook/blenderbot-400M-distill")
+
+converse = None
+
+def model_generate(text, state = None):
+    if state == None:
+        converse = Conversation(text)
+    else:
+        converse.add_user_input(text)
+
+    converse = chatbot(converse)
+    return converse.generated_responses[-1]
+
+def get_response(text, history) -> str:
+    output = model_generate(text, converse)
+    return output
+
+interface = gr.ChatInterface(get_response)
+interface.launch()

requirements.txt

@@ -0,0 +1,2 @@
+torch
+transformers