Update app.py
Browse files
app.py
CHANGED
@@ -201,11 +201,14 @@ def chat(message, history, file=None, temperature=0.9):
|
|
201 |
{"role": "user", "parts": [combined_message]}
|
202 |
]
|
203 |
|
204 |
-
#
|
205 |
-
model.generation_config
|
206 |
-
|
207 |
-
|
208 |
-
|
|
|
|
|
|
|
209 |
|
210 |
# Devolver en el formato esperado por gr.Chatbot
|
211 |
return history + [{"role": "user", "content": message}, {"role": "assistant", "content": response.text}]
|
|
|
201 |
{"role": "user", "parts": [combined_message]}
|
202 |
]
|
203 |
|
204 |
+
# Generar la respuesta - AQUÍ ESTÁ EL CAMBIO
|
205 |
+
# En lugar de modificar model.generation_config, pasamos la temperatura como parámetro
|
206 |
+
generation_config = {
|
207 |
+
"temperature": temperature,
|
208 |
+
"top_p": 1,
|
209 |
+
"max_output_tokens": 2048,
|
210 |
+
}
|
211 |
+
response = model.generate_content(messages, generation_config=generation_config)
|
212 |
|
213 |
# Devolver en el formato esperado por gr.Chatbot
|
214 |
return history + [{"role": "user", "content": message}, {"role": "assistant", "content": response.text}]
|