Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -35,31 +35,6 @@ db = get_db_firestore()
|
|
35 |
# create ASR ML pipeline
|
36 |
asr = pipeline("automatic-speech-recognition", "facebook/wav2vec2-base-960h")
|
37 |
|
38 |
-
MODEL_NAMES = [
|
39 |
-
"en/ljspeech/tacotron2-DDC",
|
40 |
-
"en/ljspeech/glow-tts",
|
41 |
-
"en/ljspeech/speedy-speech-wn",
|
42 |
-
"en/ljspeech/vits",
|
43 |
-
"en/sam/tacotron-DDC",
|
44 |
-
"fr/mai/tacotron2-DDC",
|
45 |
-
"de/thorsten/tacotron2-DCA",
|
46 |
-
]
|
47 |
-
MODELS = {}
|
48 |
-
manager = ModelManager()
|
49 |
-
for MODEL_NAME in MODEL_NAMES:
|
50 |
-
print(f"downloading {MODEL_NAME}")
|
51 |
-
model_path, config_path, model_item = manager.download_model(f"tts_models/{MODEL_NAME}")
|
52 |
-
vocoder_name: Optional[str] = model_item["default_vocoder"]
|
53 |
-
vocoder_path = None
|
54 |
-
vocoder_config_path = None
|
55 |
-
if vocoder_name is not None:
|
56 |
-
vocoder_path, vocoder_config_path, _ = manager.download_model(vocoder_name)
|
57 |
-
|
58 |
-
synthesizer = Synthesizer(
|
59 |
-
model_path, config_path, None, vocoder_path, vocoder_config_path,
|
60 |
-
)
|
61 |
-
MODELS[MODEL_NAME] = synthesizer
|
62 |
-
|
63 |
# create Text Classification pipeline
|
64 |
classifier = pipeline("text-classification")
|
65 |
|
|
|
35 |
# create ASR ML pipeline
|
36 |
asr = pipeline("automatic-speech-recognition", "facebook/wav2vec2-base-960h")
|
37 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
38 |
# create Text Classification pipeline
|
39 |
classifier = pipeline("text-classification")
|
40 |
|