Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -15,12 +15,64 @@ models = [
|
|
| 15 |
['mms fa','https://huggingface.co/willwade/mms-tts-multilingual-models-onnx/resolve/main/fas'],
|
| 16 |
['coqui-vits-female1-karim23657','https://huggingface.co/karim23657/persian-tts-vits/tree/main/persian-tts-female1-vits-coqui'],
|
| 17 |
['coqui-vits-male1-karim23657','https://huggingface.co/karim23657/persian-tts-vits/tree/main/persian-tts-male1-vits-coqui'],
|
|
|
|
|
|
|
|
|
|
|
|
|
| 18 |
['vits-piper-fa-gyro-medium','https://github.com/k2-fsa/sherpa-onnx/releases/download/tts-models/vits-piper-fa_IR-gyro-medium.tar.bz2'],
|
| 19 |
['piper-fa-amir-medium','https://github.com/k2-fsa/sherpa-onnx/releases/download/tts-models/vits-piper-fa_IR-amir-medium.tar.bz2'],
|
| 20 |
['vits-mimic3-fa-haaniye_low','https://github.com/k2-fsa/sherpa-onnx/releases/download/tts-models/vits-mimic3-fa-haaniye_low.tar.bz2'],
|
| 21 |
-
|
| 22 |
]
|
| 23 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 24 |
def download_and_extract_model(url, destination):
|
| 25 |
"""Download and extract the model files."""
|
| 26 |
print(f"Downloading from URL: {url}")
|
|
@@ -220,9 +272,9 @@ def generate_audio(text, model_info):
|
|
| 220 |
if not os.path.exists(model_dir):
|
| 221 |
print(f"Model directory doesn't exist, downloading {model_info}...")
|
| 222 |
os.makedirs(model_dir, exist_ok=True)
|
| 223 |
-
for i in
|
| 224 |
-
if model_info == i[
|
| 225 |
-
model_url=i[
|
| 226 |
download_and_extract_model(model_url, model_dir)
|
| 227 |
|
| 228 |
print(f"Contents of {model_dir}:")
|
|
|
|
| 15 |
['mms fa','https://huggingface.co/willwade/mms-tts-multilingual-models-onnx/resolve/main/fas'],
|
| 16 |
['coqui-vits-female1-karim23657','https://huggingface.co/karim23657/persian-tts-vits/tree/main/persian-tts-female1-vits-coqui'],
|
| 17 |
['coqui-vits-male1-karim23657','https://huggingface.co/karim23657/persian-tts-vits/tree/main/persian-tts-male1-vits-coqui'],
|
| 18 |
+
['coqui-vits-male-karim23657','https://huggingface.co/karim23657/persian-tts-vits/tree/main/male-male-coqui-vits'],
|
| 19 |
+
['coqui-vits-female-karim23657','https://huggingface.co/karim23657/persian-tts-vits/tree/main/female-female-coqui-vits'],
|
| 20 |
+
['coqui-vits-female-GPTInformal-karim23657','https://huggingface.co/karim23657/persian-tts-vits/tree/main/female-GPTInformal-coqui-vits'],
|
| 21 |
+
['coqui-vits-male-SmartGitiCorp','https://huggingface.co/karim23657/persian-tts-vits/tree/main/male-SmartGitiCorp-coqui-vits'],
|
| 22 |
['vits-piper-fa-gyro-medium','https://github.com/k2-fsa/sherpa-onnx/releases/download/tts-models/vits-piper-fa_IR-gyro-medium.tar.bz2'],
|
| 23 |
['piper-fa-amir-medium','https://github.com/k2-fsa/sherpa-onnx/releases/download/tts-models/vits-piper-fa_IR-amir-medium.tar.bz2'],
|
| 24 |
['vits-mimic3-fa-haaniye_low','https://github.com/k2-fsa/sherpa-onnx/releases/download/tts-models/vits-mimic3-fa-haaniye_low.tar.bz2'],
|
| 25 |
+
['vits-piper-fa_en-rezahedayatfar-ibrahimwalk-medium','https://github.com/k2-fsa/sherpa-onnx/releases/download/tts-models/vits-piper-fa_en-rezahedayatfar-ibrahimwalk-medium.tar.bz2'],
|
| 26 |
]
|
| 27 |
+
models_n =[
|
| 28 |
+
[
|
| 29 |
+
"mms fa",
|
| 30 |
+
"🌠 راد"
|
| 31 |
+
],
|
| 32 |
+
[
|
| 33 |
+
"coqui-vits-female1-karim23657",
|
| 34 |
+
"🌺 نگار"
|
| 35 |
+
],
|
| 36 |
+
[
|
| 37 |
+
"coqui-vits-male1-karim23657",
|
| 38 |
+
"🌟 آرش"
|
| 39 |
+
],
|
| 40 |
+
[
|
| 41 |
+
"coqui-vits-male-karim23657",
|
| 42 |
+
"🦁 کیان"
|
| 43 |
+
],
|
| 44 |
+
[
|
| 45 |
+
"coqui-vits-female-karim23657",
|
| 46 |
+
"🌷 مهتاب"
|
| 47 |
+
],
|
| 48 |
+
[
|
| 49 |
+
"coqui-vits-female-GPTInformal-karim23657",
|
| 50 |
+
"🌼 شیوا"
|
| 51 |
+
],
|
| 52 |
+
[
|
| 53 |
+
"coqui-vits-male-SmartGitiCorp",
|
| 54 |
+
"🚀 بهمن"
|
| 55 |
+
],
|
| 56 |
+
[
|
| 57 |
+
"vits-piper-fa-gyro-medium",
|
| 58 |
+
"💧 نیما"
|
| 59 |
+
],
|
| 60 |
+
[
|
| 61 |
+
"piper-fa-amir-medium",
|
| 62 |
+
"⚡️ آریا"
|
| 63 |
+
],
|
| 64 |
+
[
|
| 65 |
+
"vits-mimic3-fa-haaniye_low",
|
| 66 |
+
"🌹 ریما"
|
| 67 |
+
],
|
| 68 |
+
[
|
| 69 |
+
"vits-piper-fa_en-rezahedayatfar-ibrahimwalk-medium",
|
| 70 |
+
"🌠 پیام"
|
| 71 |
+
]
|
| 72 |
+
]
|
| 73 |
+
for i,m in enumerate(models):
|
| 74 |
+
models_n[i][0]=m[1]
|
| 75 |
+
dropdown_choices = list([i[1] for i in models_n])
|
| 76 |
def download_and_extract_model(url, destination):
|
| 77 |
"""Download and extract the model files."""
|
| 78 |
print(f"Downloading from URL: {url}")
|
|
|
|
| 272 |
if not os.path.exists(model_dir):
|
| 273 |
print(f"Model directory doesn't exist, downloading {model_info}...")
|
| 274 |
os.makedirs(model_dir, exist_ok=True)
|
| 275 |
+
for i in models_n:
|
| 276 |
+
if model_info == i[1]:
|
| 277 |
+
model_url=i[0]
|
| 278 |
download_and_extract_model(model_url, model_dir)
|
| 279 |
|
| 280 |
print(f"Contents of {model_dir}:")
|