aditii09 commited on
Commit
4db0c91
1 Parent(s): ec2699a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -11
app.py CHANGED
@@ -10,10 +10,7 @@ processor = Wav2Vec2Processor.from_pretrained("aditii09/facebook_english_asr")
10
  model = Wav2Vec2ForCTC.from_pretrained("aditii09/facebook_english_asr")
11
 
12
  def speech2text(audio):
13
- if isinstance(audio, tuple) and len(audio) == 2:
14
- sr, data = audio
15
- else:
16
- raise ValueError("Invalid audio format. Expected (sr, data) tuple.")
17
 
18
  # resample to 16hz
19
  data_16hz = librosa.resample(data[:,0].astype(np.float32),sr,16000)
@@ -30,11 +27,6 @@ def speech2text(audio):
30
 
31
  return transcription[0].lower() # batch size 1
32
 
33
- iface = gr.Interface(
34
- fn=speech2text,
35
- inputs=gr.Audio(source="upload", type="filepath"),
36
- examples=[["example.flac"]],
37
- outputs="text"
38
- )
39
 
40
- iface.launch()
 
10
  model = Wav2Vec2ForCTC.from_pretrained("aditii09/facebook_english_asr")
11
 
12
  def speech2text(audio):
13
+ sr, data = audio
 
 
 
14
 
15
  # resample to 16hz
16
  data_16hz = librosa.resample(data[:,0].astype(np.float32),sr,16000)
 
27
 
28
  return transcription[0].lower() # batch size 1
29
 
30
+ iface = gr.Interface(speech2text, "upload", "text")
 
 
 
 
 
31
 
32
+ iface.launch()