SergioSCA commited on
Commit
155b0ae
·
verified ·
1 Parent(s): c88db60

Update functions.py

Browse files
Files changed (1) hide show
  1. functions.py +2 -1
functions.py CHANGED
@@ -2,6 +2,7 @@
2
  import json
3
  import requests
4
  from datasets import load_dataset
 
5
 
6
  # Cargar modalidades y tareas desde un archivo JSON
7
  def cargar_modalidades_tareas():
@@ -89,7 +90,7 @@ def generar_encabezado_inteligente(tareas_seleccionadas, 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
 
2
  import json
3
  import requests
4
  from datasets import load_dataset
5
+ import matplotlib.pyplot as plt # Importar matplotlib para gráficos
6
 
7
  # Cargar modalidades y tareas desde un archivo JSON
8
  def cargar_modalidades_tareas():
 
90
  raise ValueError("Debes proporcionar una API Key válida.")
91
 
92
  # Usar un modelo de HuggingFace para generar el encabezado
93
+ API_URL = "https://api-inference.huggingface.co/models/google/flan-t5-large" # Modelo alternativo
94
  headers = {"Authorization": f"Bearer {api_key}"} # Usar la API Key proporcionada por el usuario
95
 
96
  # Construir el prompt para el modelo