Update app.py

app.py

@@ -11,25 +11,32 @@ tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)
 
 # Function to generate text
-def generate_text(prompt, max_new_tokens=100):
+def generate_text(prompt, max_new_tokens=200):
     inputs = tokenizer(prompt, return_tensors="pt")
     outputs = model.generate(
         inputs.input_ids,
         max_new_tokens=max_new_tokens,
         do_sample=True,
+        top_k=50,  # Consider only the top 50 tokens for sampling
         top_p=0.95,
-        temperature=0.7,
+        temperature=0.8,  # Increase diversity slightly
         pad_token_id=tokenizer.eos_token_id  # Ensure correct behavior for padding
     )
     text = tokenizer.decode(outputs[0], skip_special_tokens=True, clean_up_tokenization_spaces=True)
     return text
 
 # Function to generate situations and copy according to user selections
-def generate_situations_and_copys(stage, product, situation):
+def generate_situations_and_copies(stage, product, situation):
     prompt = f"""
     Estás creando una campaña publicitaria para la etapa {stage} del funnel de marketing. El producto es: {product}.
     La situación cotidiana es: {situation}. Integra el concepto de marca "Tú eliges" en el copy.
-    Genera 5 opciones de copy (titular y bajada):
+    
+    Genera 5 opciones de copy (titular y bajada) en español:
+    1.
+    2.
+    3.
+    4.
+    5.
     """
     return generate_text(prompt)
 
@@ -61,7 +68,7 @@ def update_dropdowns(stage):
     return gr.Dropdown.update(choices=product_choices), gr.Dropdown.update(choices=situation_choices)
 
 with gr.Blocks() as demo:
-    gr.Markdown("# Herramienta de Generación de Copys Publicitarios")
+    gr.Markdown("# Herramienta de Generación de Copies Publicitarios")
 
     with gr.Row():
         stage_input = gr.Dropdown(choices=stages, label="Etapa del Funnel", value="Upper")
@@ -70,14 +77,14 @@ with gr.Blocks() as demo:
 
     stage_input.change(fn=update_dropdowns, inputs=stage_input, outputs=[product_input, situation_input])
 
-    generate_button = gr.Button("Generar Copys")
-    output_text = gr.Textbox(label="Copys Generados")
+    generate_button = gr.Button("Generar Copies")
+    output_text = gr.Textbox(label="Copies Generados")
 
-    generate_button.click(fn=generate_situations_and_copys, inputs=[stage_input, product_input, situation_input], outputs=output_text)
+    generate_button.click(fn=generate_situations_and_copies, inputs=[stage_input, product_input, situation_input], outputs=output_text)
 
     custom_prompt_input = gr.Textbox(label="Ingresa un prompt para personalizar el copy", value="")
-    generate_custom_button = gr.Button("Generar Copys Personalizados")
-    custom_output_text = gr.Textbox(label="Copys Personalizados")
+    generate_custom_button = gr.Button("Generar Copies Personalizados")
+    custom_output_text = gr.Textbox(label="Copies Personalizados")
 
     generate_custom_button.click(fn=generate_text, inputs=custom_prompt_input, outputs=custom_output_text)