|
import gradio as gr |
|
import gemini_gradio |
|
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] |
|
|
|
with gr.Blocks() as demo: |
|
system_prompt = gr.Textbox("You are helpful AI.", label="System Prompt") |
|
slider = gr.Slider(10, 100, render=False) |
|
|
|
|
|
gemini_interface = gr.load( |
|
name="gemini-1.5-flash", |
|
src=gemini_gradio.registry, |
|
fill_height=True, |
|
chatbot=gr.Chatbot(type="messages") |
|
) |
|
|
|
|
|
|
|
|
|
demo.launch(ssr_mode=False) |
|
|