Spaces:
Sleeping
Sleeping
Update sugerencias_nlp.py
Browse files- sugerencias_nlp.py +3 -2
sugerencias_nlp.py
CHANGED
@@ -1,6 +1,7 @@
|
|
|
|
1 |
import requests
|
2 |
|
3 |
-
HUGGINGFACE_API_TOKEN = "
|
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"
|
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()
|