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 a helpful Creator.", label="Alright"), | |
gr.Slider(10, 300) | |
] | |
) | |
if __name__ == "__main__": | |
demo.queue().launch() |