lihaoxin2020 commited on
Commit
7d5a2b7
·
1 Parent(s): 87b1d37

vits testing

Browse files
Files changed (1) hide show
  1. app.py +13 -5
app.py CHANGED
@@ -574,7 +574,7 @@ def synthandreturn(text):
574
  pass
575
  # Get two random models
576
  # mdl1, mdl2 = random.sample(list(AVAILABLE_MODELS.keys()), 2)
577
- mdl1, mdl2 = "eleven1", "eleven2"
578
  log_text(text)
579
  print("[debug] Using", mdl1, mdl2)
580
  def predict_and_update_result(text, model, result_storage):
@@ -583,11 +583,19 @@ def synthandreturn(text):
583
  result = router.predict(text, AVAILABLE_MODELS[model].lower(), api_name="/synthesize")
584
  else:
585
  # result = router.predict(text, model.lower(), api_name="/synthesize")
 
 
 
 
 
586
  result = router.predict(
587
- text,
588
- "Rachel",
589
- fn_index=0
590
- )
 
 
 
591
  except:
592
  raise gr.Error('Unable to call API, please try again :)')
593
  print('Done with', model)
 
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):
 
583
  result = router.predict(text, AVAILABLE_MODELS[model].lower(), api_name="/synthesize")
584
  else:
585
  # result = router.predict(text, model.lower(), api_name="/synthesize")
586
+ # result = router.predict(
587
+ # 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)