import gradio as gr | |
from transformers import pipeline | |
chatbot = pipeline("conversational", model="dkleczek/bert-base-polish-cased-v1") | |
def chat_with_bot(message): | |
return chatbot(message)[0]['generated_text'] | |
gr.Interface(fn=chat_with_bot, inputs="text", outputs="text").launch() | |