Update functions.py
Browse files- functions.py +6 -2
functions.py
CHANGED
|
@@ -80,13 +80,17 @@ def generar_dataset(encabezado, datasets_seleccionados, pagina_actual=1, filas_p
|
|
| 80 |
return contenido_csv
|
| 81 |
|
| 82 |
# Funci贸n inteligente para generar el encabezado usando IA de HuggingFace
|
| 83 |
-
def generar_encabezado_inteligente(tareas_seleccionadas):
|
| 84 |
if not tareas_seleccionadas:
|
| 85 |
raise ValueError("Debes seleccionar al menos una tarea.")
|
| 86 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 87 |
# Usar un modelo de HuggingFace para generar el encabezado
|
| 88 |
API_URL = "https://api-inference.huggingface.co/models/gpt-3.5-turbo"
|
| 89 |
-
headers = {"Authorization": "Bearer
|
| 90 |
|
| 91 |
# Construir el prompt para el modelo
|
| 92 |
prompt = (
|
|
|
|
| 80 |
return contenido_csv
|
| 81 |
|
| 82 |
# Funci贸n inteligente para generar el encabezado usando IA de HuggingFace
|
| 83 |
+
def generar_encabezado_inteligente(tareas_seleccionadas, api_key):
|
| 84 |
if not tareas_seleccionadas:
|
| 85 |
raise ValueError("Debes seleccionar al menos una tarea.")
|
| 86 |
|
| 87 |
+
# Validar que se haya proporcionado una API Key
|
| 88 |
+
if not api_key:
|
| 89 |
+
raise ValueError("Debes proporcionar una API Key v谩lida.")
|
| 90 |
+
|
| 91 |
# Usar un modelo de HuggingFace para generar el encabezado
|
| 92 |
API_URL = "https://api-inference.huggingface.co/models/gpt-3.5-turbo"
|
| 93 |
+
headers = {"Authorization": f"Bearer {api_key}"} # Usar la API Key proporcionada por el usuario
|
| 94 |
|
| 95 |
# Construir el prompt para el modelo
|
| 96 |
prompt = (
|