karim23657 commited on
Commit
d8619df
·
verified ·
1 Parent(s): 65067ea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +57 -5
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
- dropdown_choices = list([i[0] for i in models])
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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 models:
224
- if model_info == i[0]:
225
- model_url=i[1]
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}:")