hcsolakoglu commited on
Commit
43e252d
·
verified ·
1 Parent(s): cff403b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -5,12 +5,12 @@ from f5_tts.api import F5TTS
5
  from huggingface_hub import hf_hub_download
6
  import os
7
 
8
- VARIANTS = ["model_1000000"]
9
  models = {}
10
  for variant in VARIANTS:
11
  models[variant] = F5TTS(
12
- ckpt_file=hf_hub_download("mrfakename/openf5-v2", f"{variant}.pt", token=os.getenv("HF_TOKEN")),
13
- vocab_file=hf_hub_download("mrfakename/openf5-v2", "vocab.txt", token=os.getenv("HF_TOKEN"))
14
  )
15
 
16
  @spaces.GPU
@@ -28,7 +28,7 @@ def generate(text, ref_audio, variant, progress = gr.Progress()):
28
  return f.name
29
 
30
  with gr.Blocks() as demo:
31
- gr.Markdown("# Demo for OpenF5 TTS\n\nTry out [OpenF5-TTS](https://huggingface.co/mrfakename/OpenF5-TTS)")
32
  textbox = gr.Textbox(label="Text")
33
  audio = gr.Audio(label="Reference Audio", type="filepath")
34
  variant = gr.Radio(choices=VARIANTS, value=VARIANTS[0], label="Variant")
 
5
  from huggingface_hub import hf_hub_download
6
  import os
7
 
8
+ VARIANTS = ["orkhon_tts"]
9
  models = {}
10
  for variant in VARIANTS:
11
  models[variant] = F5TTS(
12
+ ckpt_file=hf_hub_download("hcsolakoglu/Orkhon-TTS", f"{variant}.pt", token=os.getenv("HF_TOKEN")),
13
+ vocab_file=hf_hub_download("hcsolakoglu/Orkhon-TTS", "vocab.txt", token=os.getenv("HF_TOKEN"))
14
  )
15
 
16
  @spaces.GPU
 
28
  return f.name
29
 
30
  with gr.Blocks() as demo:
31
+ gr.Markdown("# Demo for Orkhon-TTS\n\nTry out [Orkhon-TTS](https://huggingface.co/hcsolakoglu/Orkhon-TTS)")
32
  textbox = gr.Textbox(label="Text")
33
  audio = gr.Audio(label="Reference Audio", type="filepath")
34
  variant = gr.Radio(choices=VARIANTS, value=VARIANTS[0], label="Variant")