Norphel commited on
Commit
60f1c75
·
verified ·
1 Parent(s): c6d0138

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -1,7 +1,6 @@
1
  import numpy as np
2
  import torch
3
  import gradio as gr
4
- import librosa
5
  from transformers import Wav2Vec2ForCTC, Wav2Vec2Processor
6
 
7
  # Load ASR model & processor
@@ -16,6 +15,9 @@ asr_model.to(device)
16
 
17
  # Function to process audio & generate text
18
  def generate_text(audio):
 
 
 
19
  sr, data = audio
20
  data = data.astype(np.float32)
21
  input_dict = asr_processor(data, sampling_rate=16_000, return_tensors="pt", padding=True)
 
1
  import numpy as np
2
  import torch
3
  import gradio as gr
 
4
  from transformers import Wav2Vec2ForCTC, Wav2Vec2Processor
5
 
6
  # Load ASR model & processor
 
15
 
16
  # Function to process audio & generate text
17
  def generate_text(audio):
18
+ if audio is None:
19
+ return "No audio received"
20
+
21
  sr, data = audio
22
  data = data.astype(np.float32)
23
  input_dict = asr_processor(data, sampling_rate=16_000, return_tensors="pt", padding=True)