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