File size: 578 Bytes
814c07e
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
// 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`;