Update app.py

app.py

@@ -71,7 +71,7 @@ async def improve_prompt(prompt):
     try:
         instruction = ("With this idea, describe in English a detailed img2vid prompt in a single paragraph of up to 200 characters maximun, developing atmosphere, characters, lighting, and cameras.")
         formatted_prompt = f"{prompt}: {instruction}"
-        response = llm_client.text_generation(formatted_prompt, max_new_tokens=300, language="english")
+        response = llm_client.text_generation(formatted_prompt, max_new_tokens=200)
         improved_text = response['generated_text'].strip() if 'generated_text' in response else response.strip()
 
         return improved_text
@@ -99,27 +99,23 @@ with gr.Blocks(css=css, theme="Nymbo/Nymbo_Theme") as demo:
                     choices=["Shakker-Labs/FLUX.1-dev-LoRA-add-details", "XLabs-AI/flux-RealismLora"],
                     value="XLabs-AI/flux-RealismLora"
                 )
-
-            with gr.Row():
-                process_lora = gr.Checkbox(label="Procesar LORA")
-                process_upscale = gr.Checkbox(label="Procesar Escalador")
-
-            improved_prompt = gr.Textbox(label="Prompt Mejorado", interactive=False)
-            improve_btn = gr.Button("Mejora mi prompt")
-            improve_btn.click(fn=improve_prompt, inputs=[prompt], outputs=improved_prompt)
-
-            with gr.Accordion(label="Opciones Avanzadas", open=False):
-                width = gr.Slider(label="Ancho", minimum=512, maximum=1280, step=8, value=1280)
-                height = gr.Slider(label="Alto", minimum=512, maximum=1280, step=8, value=768)
-                upscale_factor = gr.Radio(label="Factor de Escala", choices=[2, 4, 8], value=2)
-                scales = gr.Slider(label="Escalado", minimum=1, maximum=20, step=1, value=10)
-                steps = gr.Slider(label="Pasos", minimum=1, maximum=100, step=1, value=20)
-                seed = gr.Number(label="Semilla", value=-1)
-
-            btn = gr.Button("Generar")
-            btn.click(
-                fn=gen, 
-                inputs=[prompt, basemodel_choice, width, height, scales, steps, seed, upscale_factor, process_upscale, lora_model_choice, process_lora], 
-                outputs=output_res
-            )
+                with gr.Row():
+                    process_lora = gr.Checkbox(label="Procesar LORA")
+                    process_upscale = gr.Checkbox(label="Procesar Escalador")
+                improved_prompt = gr.Textbox(label="Prompt Mejorado", interactive=False)
+                improve_btn = gr.Button("Mejora mi prompt")
+                improve_btn.click(fn=improve_prompt, inputs=[prompt], outputs=improved_prompt)
+                with gr.Accordion(label="Opciones Avanzadas", open=False):
+                    width = gr.Slider(label="Ancho", minimum=512, maximum=1280, step=8, value=1280)
+                    height = gr.Slider(label="Alto", minimum=512, maximum=1280, step=8, value=768)
+                    upscale_factor = gr.Radio(label="Factor de Escala", choices=[2, 4, 8], value=2)
+                    scales = gr.Slider(label="Escalado", minimum=1, maximum=20, step=1, value=10)
+                    steps = gr.Slider(label="Pasos", minimum=1, maximum=100, step=1, value=20)
+                    seed = gr.Number(label="Semilla", value=-1)
+                btn = gr.Button("Generar")
+                btn.click(
+                    fn=gen, 
+                    inputs=[prompt, basemodel_choice, width, height, scales, steps, seed, upscale_factor, process_upscale, lora_model_choice, process_lora], 
+                    outputs=output_res
+                )
     demo.launch()