| import gradio as gr | |
| import time | |
| def echo(message, history, system_prompt, tokens): | |
| response = f"System prompt: {system_prompt}\n Message: {message}." | |
| for i in range(min(len(response), int(tokens))): | |
| time.sleep(0.05) | |
| yield response[: i + 1] | |
| demo = gr.ChatInterface( | |
| echo, | |
| additional_inputs=[ | |
| gr.Textbox("You are helpful AI.", label="System Prompt"), | |
| gr.Slider(10, 100), | |
| ], | |
| ) | |
| if __name__ == "__main__": | |
| demo.launch() | |