"""Wrapper to run transkun with proper CPU device handling."""
import sys
import torch

# Patch torch.load BEFORE any transkun imports
_orig_load = torch.load
def _safe_load(*args, **kwargs):
    if 'map_location' not in kwargs:
        kwargs['map_location'] = 'cpu'
    return _orig_load(*args, **kwargs)
torch.load = _safe_load

# Patch torch.cuda.is_available to return False (force CPU)
torch.cuda.is_available = lambda: False

# Now import and run transkun's main
from transkun.transcribe import main
sys.argv = ['transkun', sys.argv[1], sys.argv[2], '--device', 'cpu']
main()