ChandJain commited on
Commit
cb5c564
1 Parent(s): 896148c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -6
app.py CHANGED
@@ -7,17 +7,15 @@ from gtts import gTTS
7
  import IPython.display as ipd
8
 
9
  #Audio to text
10
- #asr = pipeline(task="automatic-speech-recognition",
11
- # model="distil-whisper/distil-small.en")
12
- asr=pipeline("automatic-speech-recognition",
13
- model="openai/whisper-large-v3")
14
  #Text to text
15
  translator = pipeline(task="translation",
16
  model="facebook/nllb-200-distilled-600M",
17
  torch_dtype=torch.bfloat16)
18
  #Text to audio
19
- pipe = pipeline("text-to-speech", model="suno/bark-small",
20
- torch_dtype=torch.bfloat16)
21
 
22
 
23
  demo = gr.Blocks()
 
7
  import IPython.display as ipd
8
 
9
  #Audio to text
10
+ asr = pipeline(task="automatic-speech-recognition",
11
+ model="distil-whisper/distil-small.en")
12
+ #asr=pipeline("automatic-speech-recognition", model="openai/whisper-large-v3")
 
13
  #Text to text
14
  translator = pipeline(task="translation",
15
  model="facebook/nllb-200-distilled-600M",
16
  torch_dtype=torch.bfloat16)
17
  #Text to audio
18
+ pipe = pipeline("text-to-speech", model="suno/bark-small")
 
19
 
20
 
21
  demo = gr.Blocks()