copywriter / app.py
JeCabrera's picture
Update app.py
2ca19cf verified
raw
history blame
971 Bytes
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)
# Cargar la interfaz gemini sin el parámetro 'echo'
gemini_interface = gr.load(
name="gemini-1.5-flash", # Se establece el modelo por defecto
src=gemini_gradio.registry,
fill_height=True,
chatbot=gr.Chatbot(type="messages") # El Chatbot ahora no tiene el 'echo'
)
# El parámetro 'echo' debería manejarse de forma separada o a través de un proceso diferente
# Si es necesario, puedes agregar 'echo' como un generador o un proceso posterior al input
demo.launch(ssr_mode=False)