seba3y commited on
Commit
5f4b99f
1 Parent(s): 86a5e7d

Upload 2 files

Browse files
Files changed (2) hide show
  1. logic.py +4 -2
  2. requirements.txt +1 -1
logic.py CHANGED
@@ -1,11 +1,13 @@
1
  from phonemizer.separator import Separator
2
- from phonemizer import phonemize
3
  # from phonemizer.backend.espeak.wrapper import EspeakWrapper
4
  from Levenshtein import distance as levenshtein_distance
5
 
6
  import whisper
7
  import torch
8
-
 
 
9
  device = 'cuda:0' if torch.cuda.is_available() else 'cpu'
10
 
11
  model = whisper.load_model("base.en", device=device)
 
1
  from phonemizer.separator import Separator
2
+ from phonemizer import phonemize, backend
3
  # from phonemizer.backend.espeak.wrapper import EspeakWrapper
4
  from Levenshtein import distance as levenshtein_distance
5
 
6
  import whisper
7
  import torch
8
+ if not backend.EspeakBackend.is_available():
9
+ import os
10
+ os.system('sudo apt-get install espeak-ng')
11
  device = 'cuda:0' if torch.cuda.is_available() else 'cpu'
12
 
13
  model = whisper.load_model("base.en", device=device)
requirements.txt CHANGED
@@ -1,9 +1,9 @@
1
- espeak-ng
2
  phonemizer
3
  wave
4
  torch
5
  openai-whisper
6
  gradio
 
7
  numpy
8
  resampy
9
  Levenshtein
 
 
1
  phonemizer
2
  wave
3
  torch
4
  openai-whisper
5
  gradio
6
+ scipy
7
  numpy
8
  resampy
9
  Levenshtein