barghavani commited on
Commit
ab9a8c2
·
1 Parent(s): 26874b9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -7
app.py CHANGED
@@ -1,4 +1,3 @@
1
-
2
  import os
3
  import tempfile
4
  import gradio as gr
@@ -9,20 +8,24 @@ import json
9
 
10
  # Define constants
11
  MODEL_INFO = [
12
- #["vits checkpoint 57000", "checkpoint_57000.pth", "config.json", "mhrahmani/persian-tts-vits-0"],
13
- ["VITS Grapheme Multispeaker CV15(reduct)(best at 17864)", "best_model_17864.pth", "config.json", "saillab/persian-tts-cv15-reduct-grapheme-multispeaker"],
 
14
  ["VITS Grapheme Multispeaker CV15(reduct)(22000)", "checkpoint_22000.pth", "config.json", "saillab/persian-tts-cv15-reduct-grapheme-multispeaker", "speakers.pth"],
15
  ["VITS Grapheme Multispeaker CV15(reduct)(26000)", "checkpoint_25000.pth", "config.json", "saillab/persian-tts-cv15-reduct-grapheme-multispeaker", "speakers.pth"],
16
- ["vits-multispeaker-retrain-train set-90K", "best_model_56960.pth", "config.json", "saillab/multi_speaker","speakers.pth"],
17
 
18
- ["VITS Grapheme Azure (best at 15934)", "best_model_15934.pth", "config.json","saillab/persian-tts-azure-grapheme-60K"],
 
19
  ["VITS Grapheme Azure (61000)", "checkpoint_61000.pth", "config.json", "saillab/persian-tts-azure-grapheme-60K"],
20
 
21
- ["VITS Grapheme ARM24 Fine-Tuned on 1 (66651)", "best_model_66651.pth", "config.json","saillab/persian-tts-grapheme-arm24-finetuned-on1"],
22
- ["VITS Grapheme ARM24 Fine-Tuned on 1 (120000)", "checkpoint_120000.pth", "config.json","saillab/persian-tts-grapheme-arm24-finetuned-on1"],
 
 
23
 
24
  # ... Add other models similarly
25
  ]
 
26
  # Extract model names from MODEL_INFO
27
  MODEL_NAMES = [info[0] for info in MODEL_INFO]
28
 
@@ -156,3 +159,4 @@ iface = gr.Interface(
156
  )
157
 
158
  iface.launch()
 
 
 
1
  import os
2
  import tempfile
3
  import gradio as gr
 
8
 
9
  # Define constants
10
  MODEL_INFO = [
11
+ ["vits checkpoint 57000", "checkpoint_57000.pth", "config.json", "mhrahmani/persian-tts-vits-0"],
12
+ # ["VITS Grapheme Multispeaker CV15(reduct)(best at 17864)", "best_model_17864.pth", "config.json",
13
+ # "saillab/persian-tts-cv15-reduct-grapheme-multispeaker"],
14
  ["VITS Grapheme Multispeaker CV15(reduct)(22000)", "checkpoint_22000.pth", "config.json", "saillab/persian-tts-cv15-reduct-grapheme-multispeaker", "speakers.pth"],
15
  ["VITS Grapheme Multispeaker CV15(reduct)(26000)", "checkpoint_25000.pth", "config.json", "saillab/persian-tts-cv15-reduct-grapheme-multispeaker", "speakers.pth"],
 
16
 
17
+ # ["VITS Grapheme Azure (best at 15934)", "best_model_15934.pth", "config.json",
18
+ # "saillab/persian-tts-azure-grapheme-60K"],
19
  ["VITS Grapheme Azure (61000)", "checkpoint_61000.pth", "config.json", "saillab/persian-tts-azure-grapheme-60K"],
20
 
21
+ ["VITS Grapheme ARM24 Fine-Tuned on 1 (66651)", "best_model_66651.pth", "config.json",
22
+ "saillab/persian-tts-grapheme-arm24-finetuned-on1"],
23
+ ["VITS Grapheme ARM24 Fine-Tuned on 1 (120000)", "checkpoint_120000.pth", "config.json",
24
+ "saillab/persian-tts-grapheme-arm24-finetuned-on1"],
25
 
26
  # ... Add other models similarly
27
  ]
28
+
29
  # Extract model names from MODEL_INFO
30
  MODEL_NAMES = [info[0] for info in MODEL_INFO]
31
 
 
159
  )
160
 
161
  iface.launch()
162
+