JeCabrera commited on
Commit
cf439a4
·
verified ·
1 Parent(s): 9bbefe4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -18,12 +18,12 @@ def to_markdown(text):
18
  def generate_headlines(number_of_headlines, target_audience, product, temperature):
19
  # Crear la configuración del modelo
20
  generation_config = {
21
- "temperature": temperature, # Usar el valor del slider aquí
22
- "top_p": 0.95,
23
- "top_k": 64,
24
- "max_output_tokens": 8192,
25
- "response_mime_type": "text/plain",
26
- }
27
 
28
  model = genai.GenerativeModel(
29
  model_name="gemini-1.5-pro",
 
18
  def generate_headlines(number_of_headlines, target_audience, product, temperature):
19
  # Crear la configuración del modelo
20
  generation_config = {
21
+ "temperature": temperature, # Usar el valor del slider aquí
22
+ "top_p": 0.95,
23
+ "top_k": 64,
24
+ "max_output_tokens": 150, # Limitar a 150 tokens
25
+ "response_mime_type": "text/plain",
26
+ }
27
 
28
  model = genai.GenerativeModel(
29
  model_name="gemini-1.5-pro",