import torch

# Local config
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"

# HF API config
TAG_NAMES = [
    'games',
    'geometry',
    'graphs',
    'math',
    'number theory',
    'other',
    'probabilities',
    'strings',
    'trees'
    ]