a

app.py

@@ -3,10 +3,10 @@ import asyncio
 import google.generativeai as genai
 import gradio as gr
 
-# Configuración de claves de API
+# Configuración de la clave de API
 GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
 if not GEMINI_API_KEY:
-    raise ValueError("La clave de API GEMINI_API_KEY no está configurada correctamente.")
+    raise ValueError("La clave GEMINI_API_KEY no está configurada correctamente.")
 genai.configure(api_key=GEMINI_API_KEY)
 
 # Instrucciones del sistema por defecto
@@ -16,9 +16,29 @@ Como Ministro de la Corte Suprema de Chile, especializado en Derecho de Familia,
 El objetivo es elevar el texto a un estándar de excelencia en redacción jurídica, asegurando la máxima claridad, precisión, concisión y formalidad. **No debes modificar la estructura del borrador, tampoco agregar fundamentación o hechos. La mejora solo es gramatical, redaccional y estética lingüística jurídica.**
 """
 
-# Configuración de modelos de Google
-google_gemini_model = genai.GenerativeModel("gemini-exp-1121", generation_config={...})
-google_learnlm_model = genai.GenerativeModel("learnlm-1.5-pro-experimental", generation_config={...})
+# Configuración del modelo de Google Gemini
+google_gemini_model = genai.GenerativeModel(
+    "gemini-exp-1121",
+    generation_config={
+        "temperature": 0.5,
+        "top_p": 0.9,
+        "top_k": 40,
+        "max_output_tokens": 5000,
+        "response_mime_type": "text/plain",
+    },
+)
+
+# Configuración del modelo de Google LearnLM
+google_learnlm_model = genai.GenerativeModel(
+    "learnlm-1.5-pro-experimental",
+    generation_config={
+        "temperature": 0.5,
+        "top_p": 0.9,
+        "top_k": 40,
+        "max_output_tokens": 5000,
+        "response_mime_type": "text/plain",
+    },
+)
 
 async def generate_content(client, model_name, system_instruction, borrador):
     try:
@@ -53,4 +73,4 @@ with gr.Blocks() as demo:
     submit_btn.click(fn=predict, inputs=borrador, outputs=output)
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.queue(concurrency_count=2, max_size=10).launch(timeout_seconds=3600)