needle-playground / src /config.ts
shreyask's picture
Upload folder using huggingface_hub
814c07e verified
raw
history blame contribute delete
578 Bytes
// Single config switch: dev loads from /models-dev/ (Vite-served), prod from HF Hub.
// Flip MODEL_BASE_URL when moving to a Hub-hosted production deploy (Task 16).
export const MODEL_BASE_URL = import.meta.env.PROD
? 'https://huggingface.co/onnx-community/needle-onnx/resolve/main'
: '/models-dev';
export const ENCODER_URL = `${MODEL_BASE_URL}/encoder.onnx`;
export const DECODER_URL = `${MODEL_BASE_URL}/decoder_step.onnx`;
export const TOKENIZER_URL = `${MODEL_BASE_URL}/needle.model`;
export const SPECIALS_URL = `${MODEL_BASE_URL}/tokenizer-specials.json`;