Spaces:
Sleeping
Sleeping
Upload 2 files
Browse files- logic.py +4 -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
|