mirror of
https://github.com/gradio-app/gradio.git
synced 2025-01-06 10:25:17 +08:00
18 lines
577 B
Python
18 lines
577 B
Python
|
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.queue().launch()
|