Spaces:
Build error
Build error
Update app.py
Browse files
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 |
-
|
| 22 |
-
|
| 23 |
-
|
| 24 |
-
|
| 25 |
-
|
| 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",
|