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()
|