MusIre commited on
Commit
6709e5b
·
1 Parent(s): 1dcf9a8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -16,9 +16,15 @@ forced_decoder_ids = processor.get_decoder_prompt_ids(language="italian", task="
16
 
17
 
18
  def transcribe_audio(input_audio):
19
- input_audio_np = np.array(input_audio[0].data)
 
 
 
 
 
20
  input_features = processor(input_audio_np, return_tensors="pt").input_features
21
 
 
22
  # Generate token ids
23
  predicted_ids = model.generate(input_features, forced_decoder_ids=forced_decoder_ids)
24
 
 
16
 
17
 
18
  def transcribe_audio(input_audio):
19
+ if isinstance(input_audio, int):
20
+ # Handle the case where input_audio is an integer (error fallback)
21
+ input_audio_np = np.array([0.0]) # You can adjust this default value
22
+ else:
23
+ input_audio_np = np.array(input_audio.data)
24
+
25
  input_features = processor(input_audio_np, return_tensors="pt").input_features
26
 
27
+
28
  # Generate token ids
29
  predicted_ids = model.generate(input_features, forced_decoder_ids=forced_decoder_ids)
30