Ethgoin commited on
Commit
cc67875
·
verified ·
1 Parent(s): 75eb629

Update sugerencias_nlp.py

Browse files
Files changed (1) hide show
  1. sugerencias_nlp.py +3 -2
sugerencias_nlp.py CHANGED
@@ -1,6 +1,7 @@
 
1
  import requests
2
 
3
- HUGGINGFACE_API_TOKEN = "hf_..." # Reemplazar con tu token real
4
  API_URL = "https://api-inference.huggingface.co/models/gpt2"
5
 
6
  headers = {
@@ -22,7 +23,7 @@ def obtener_sugerencia_nlp(texto_comentario):
22
  resultado = response.json()
23
  return resultado[0]["generated_text"].strip()
24
  else:
25
- return f"# (No se pudo generar sugerencia: {response.status_code})"
26
 
27
  def procesar_comentarios(codigo: str):
28
  lineas = codigo.splitlines()
 
1
+ import os
2
  import requests
3
 
4
+ HUGGINGFACE_API_TOKEN = os.environ.get("HF_TOKEN", "")
5
  API_URL = "https://api-inference.huggingface.co/models/gpt2"
6
 
7
  headers = {
 
23
  resultado = response.json()
24
  return resultado[0]["generated_text"].strip()
25
  else:
26
+ return f"(sin sugerencia: {response.status_code})"
27
 
28
  def procesar_comentarios(codigo: str):
29
  lineas = codigo.splitlines()