Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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)
|