jjvelezo commited on
Commit
1a451f5
·
verified ·
1 Parent(s): 6edd5f9

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
agent.py CHANGED
@@ -4,7 +4,7 @@ from huggingface_hub import hf_api
4
 
5
  # Cargar el token de Hugging Face desde el entorno
6
  HF_TOKEN = os.getenv("HF_TOKEN")
7
- MODEL_NAME = os.getenv("MODEL_NAME", "meta-llama/llama-3.2-3b-instruct") # Cambia al nombre del modelo si es necesario
8
  print("HF_TOKEN desde agent.py:", HF_TOKEN[:10] if HF_TOKEN else "NO DEFINIDO")
9
  # Función para llamar al modelo y obtener respuesta
10
  def call_model(prompt):
 
4
 
5
  # Cargar el token de Hugging Face desde el entorno
6
  HF_TOKEN = os.getenv("HF_TOKEN")
7
+ MODEL_NAME = os.getenv("MODEL_NAME", "google/gemma-2b-it")
8
  print("HF_TOKEN desde agent.py:", HF_TOKEN[:10] if HF_TOKEN else "NO DEFINIDO")
9
  # Función para llamar al modelo y obtener respuesta
10
  def call_model(prompt):