Manobhiram13/ASR-Telugu / example_inference.py
Manobhiram13's picture
download
raw
849 Bytes
import sys, torch
from transformers import AutoModel
REPO = "."
DEV = "cuda" if torch.cuda.is_available() else "cpu"
model = AutoModel.from_pretrained(REPO, trust_remote_code=True).to(DEV).eval()
# NeMo-style convenience API (needs sentencepiece; soundfile or stdlib wave for files):
hyps = model.transcribe(sys.argv[1:], return_hypotheses=True)
for path, h in zip(sys.argv[1:], hyps):
print(f"{path}\t{h.text}")
# --- lower-level alternative (explicit processor) ---
# from transformers import AutoProcessor
# import soundfile as sf # or: import wave (stdlib) for PCM WAV
# proc = AutoProcessor.from_pretrained(REPO, trust_remote_code=True)
# wav, sr = sf.read(path, dtype="float32") # average channels if stereo
# inputs = proc(wav, sampling_rate=sr, return_tensors="pt").to(DEV)
# text = proc.batch_decode(model.generate(**inputs))[0]

Xet Storage Details

Size:
849 Bytes
·
Xet hash:
e8a1aa92602c01cf0593303abb0153bed5cef939f2a42107c29c9ae077788bc3

Xet efficiently stores files, intelligently splitting them into unique chunks and accelerating uploads and downloads. More info.