Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
import spaces
|
2 |
-
|
3 |
import gradio as gr
|
4 |
import torch
|
5 |
import torchaudio
|
@@ -9,6 +9,7 @@ import torchaudio.transforms as T
|
|
9 |
import os
|
10 |
from pathlib import Path
|
11 |
|
|
|
12 |
class TextTransform:
|
13 |
def __init__(self):
|
14 |
char_map_str = """
|
@@ -184,8 +185,8 @@ model = SpeechRecognitionModel(
|
|
184 |
hparams['n_cnn_layers'], hparams['n_rnn_layers'], hparams['rnn_dim'],
|
185 |
hparams['n_class'], hparams['n_feats'], hparams['stride'], hparams['dropout']
|
186 |
)
|
187 |
-
model = gr.load("NeerAbhy/Speech_recognotion_model", src = 'models')
|
188 |
-
|
189 |
def prediction(audio):
|
190 |
valid_audio_transforms = torchaudio.transforms.MelSpectrogram(sample_rate = 16000)
|
191 |
|
|
|
1 |
import spaces
|
2 |
+
from huggingface_hub import from_pretrained_fastai
|
3 |
import gradio as gr
|
4 |
import torch
|
5 |
import torchaudio
|
|
|
9 |
import os
|
10 |
from pathlib import Path
|
11 |
|
12 |
+
repo_id = "NeerAbhy/Speech_recognotion_model"
|
13 |
class TextTransform:
|
14 |
def __init__(self):
|
15 |
char_map_str = """
|
|
|
185 |
hparams['n_cnn_layers'], hparams['n_rnn_layers'], hparams['rnn_dim'],
|
186 |
hparams['n_class'], hparams['n_feats'], hparams['stride'], hparams['dropout']
|
187 |
)
|
188 |
+
#model = gr.load("NeerAbhy/Speech_recognotion_model", src = 'models')
|
189 |
+
model = from_pretrained_fastai(repo_id)
|
190 |
def prediction(audio):
|
191 |
valid_audio_transforms = torchaudio.transforms.MelSpectrogram(sample_rate = 16000)
|
192 |
|