JeCabrera commited on
Commit
7186e2b
·
verified ·
1 Parent(s): e599aa5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -80,6 +80,10 @@ def generate_headlines(number_of_headlines, target_audience, product, temperatur
80
  product_mention = get_random_product_mention()
81
  mention_instruction = get_mention_instruction(product_mention, product)
82
 
 
 
 
 
83
  # Crear la configuración del modelo
84
  generation_config = {
85
  "temperature": temperature,
 
80
  product_mention = get_random_product_mention()
81
  mention_instruction = get_mention_instruction(product_mention, product)
82
 
83
+ # Seleccionar tipos de titulares
84
+ all_types = list(headline_types.keys())
85
+ selected_types = random.choices(all_types, k=number_of_headlines) # Permite repetición
86
+
87
  # Crear la configuración del modelo
88
  generation_config = {
89
  "temperature": temperature,