soiz commited on
Commit
c4007b5
·
verified ·
1 Parent(s): 23e60a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ def transcribe_audio(audio):
18
 
19
  # モデルによる文字起こし
20
  with torch.no_grad():
21
- predicted_ids = model.generate(input_ids=audio_input.input_ids)
22
 
23
  # 文字起こし結果のデコード
24
  transcription = processor.decode(predicted_ids[0], skip_special_tokens=True)
 
18
 
19
  # モデルによる文字起こし
20
  with torch.no_grad():
21
+ predicted_ids = model.generate(input_values=audio_input.input_values) # 修正: input_ids → input_values
22
 
23
  # 文字起こし結果のデコード
24
  transcription = processor.decode(predicted_ids[0], skip_special_tokens=True)