ranggaaldosas commited on
Commit
3e4a732
·
1 Parent(s): 54d342c

feat: add audio error handling

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -24,7 +24,13 @@ model = WhisperForConditionalGeneration.from_pretrained(
24
 
25
 
26
  def transcribe(audio):
27
- audio, sampling_rate = librosa.load(audio, sr=16000)
 
 
 
 
 
 
28
  audio_input = processor(audio, return_tensors="pt", sampling_rate=16000)
29
  input_values = audio_input.input_features
30
 
 
24
 
25
 
26
  def transcribe(audio):
27
+ if audio is None:
28
+ return "No audio file provided. Please upload an audio file."
29
+ try:
30
+ audio, sampling_rate = librosa.load(audio, sr=16000)
31
+ except Exception as e:
32
+ return f"Failed to load audio: {str(e)}"
33
+
34
  audio_input = processor(audio, return_tensors="pt", sampling_rate=16000)
35
  input_values = audio_input.input_features
36