Fix: Update position embedding to use idx.device

minor bug : variable name differs

deploy - production app with model weights and tokenizer