| from transformers import pipeline | |
| import textwrap | |
| import time | |
| import gradio as gr | |
| # Load a pre-trained chatbot model | |
| chatbot = pipeline("conversational") | |
| def wrap(x): | |
| return textwrap.fill(x, replace_whitespace=False, fix_sentence_endings=True) | |
| def chat(prompt): | |
| out = chatbot(prompt) | |
| for i in wrap(out[0]['generated_text']): | |
| print(i, end='') | |
| time.sleep(0.2) | |
| gr.Interface(fn=chat, inputs="text", outputs="text").launch() | |