lihaoxin2020 commited on
Commit
69b4ed2
·
1 Parent(s): 7d5a2b7

moe testing

Browse files
Files changed (1) hide show
  1. app.py +12 -10
app.py CHANGED
@@ -141,7 +141,8 @@ scheduler = CommitScheduler(
141
  # Router API
142
  ####################################
143
  # router = Client("TTS-AGI/tts-router", hf_token=os.getenv('HF_TOKEN'))
144
- router = Client("https://elevenlabs-tts.hf.space/")
 
145
  ####################################
146
  # Gradio app
147
  ####################################
@@ -574,7 +575,7 @@ def synthandreturn(text):
574
  pass
575
  # Get two random models
576
  # mdl1, mdl2 = random.sample(list(AVAILABLE_MODELS.keys()), 2)
577
- mdl1, mdl2 = "vits1", "vits2"
578
  log_text(text)
579
  print("[debug] Using", mdl1, mdl2)
580
  def predict_and_update_result(text, model, result_storage):
@@ -588,14 +589,15 @@ def synthandreturn(text):
588
  # "Rachel",
589
  # fn_index=0
590
  # )
591
- result = router.predict(
592
- language="English",
593
- repo_id="csukuangfj/vits-piper-en_US-glados|1 speaker",
594
- text=text,
595
- sid="0",
596
- speed=1,
597
- api_name="/process"
598
- )[0]
 
599
  except:
600
  raise gr.Error('Unable to call API, please try again :)')
601
  print('Done with', model)
 
141
  # Router API
142
  ####################################
143
  # router = Client("TTS-AGI/tts-router", hf_token=os.getenv('HF_TOKEN'))
144
+ # router = Client("https://elevenlabs-tts.hf.space/")
145
+ router = Client("skytnt/moe-tts")
146
  ####################################
147
  # Gradio app
148
  ####################################
 
575
  pass
576
  # Get two random models
577
  # mdl1, mdl2 = random.sample(list(AVAILABLE_MODELS.keys()), 2)
578
+ mdl1, mdl2 = "moe1", "moe2"
579
  log_text(text)
580
  print("[debug] Using", mdl1, mdl2)
581
  def predict_and_update_result(text, model, result_storage):
 
589
  # "Rachel",
590
  # fn_index=0
591
  # )
592
+ # result = router.predict(
593
+ # language="English",
594
+ # repo_id="csukuangfj/vits-piper-en_US-glados|1 speaker",
595
+ # text=text,
596
+ # sid="0",
597
+ # speed=1,
598
+ # api_name="/process"
599
+ # )[0]
600
+ result = router.predict("hello", api_name="/tts_fn_7")[1]
601
  except:
602
  raise gr.Error('Unable to call API, please try again :)')
603
  print('Done with', model)