"""Descarga el modelo y el tokenizer desde Hugging Face Hub y clasifica EN ESTA PC
(o en el servidor donde corra).

Con Transformers alcanza con el nombre del repo: pipeline() baja config.json,
model.safetensors y el tokenizer.

Uso:
    python predict_hub.py tu-usuario/clasificador-comentarios-transformers
"""

import sys

from transformers import pipeline

REPO_ID = sys.argv[1] if len(sys.argv) > 1 else "TU-USUARIO/clasificador-comentarios-transformers"

clasificador = pipeline("text-classification", model=REPO_ID)

comentarios = [
    "el departamento es muy luminoso",
    "me pareció un lugar ruidoso",
    "es amplio pero sobre todo tranquilo",
]
for comentario, resultado in zip(comentarios, clasificador(comentarios)):
    print(f"{comentario!r} -> {resultado['label']} ({resultado['score']:.2f})")