qwen_classifier/config.py

import torch

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

# HF API config
SPACE_URL = "https://keivanr-qwen-classifier-demo.hf.space"
HF_REPO = 'KeivanR/Qwen2.5-1.5B-Instruct-MLB-clf_3layer-1743180976'

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

EVAL_LIMIT = 8