SergioSCA commited on
Commit
51ecf47
verified
1 Parent(s): 7ae6cdf

Update functions.py

Browse files
Files changed (1) hide show
  1. 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 TU_API_KEY"} # Reemplaza TU_API_KEY con tu clave de HuggingFace
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 = (