MeBai commited on
Commit
a151b3f
·
verified ·
1 Parent(s): 4c1eb71

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -2
app.py CHANGED
@@ -48,10 +48,18 @@ def transcribe(audio_path):
48
 
49
 
50
  #------
51
- audio_input = processor(audio_path, return_tensors="pt", sampling_rate=16000).input_values
 
 
 
 
 
52
  with torch.no_grad():
53
- predicted_ids = model.generate(audio_input)
 
 
54
  transcription = processor.batch_decode(predicted_ids, skip_special_tokens=True)[0]
 
55
  return transcription
56
 
57
 
 
48
 
49
 
50
  #------
51
+ audio_cnt, sr = librosa.load(audio_path, sr=16000)
52
+
53
+ # 将音频数据传递给 processor
54
+ input_values = processor(audio, sampling_rate=16000, return_tensors="pt").input_values
55
+
56
+ # 模型推理
57
  with torch.no_grad():
58
+ predicted_ids = model.generate(input_values)
59
+
60
+ # 解码得到转录结果
61
  transcription = processor.batch_decode(predicted_ids, skip_special_tokens=True)[0]
62
+
63
  return transcription
64
 
65