NeerAbhy commited on
Commit
d4013a9
·
verified ·
1 Parent(s): 68cee16

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
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