amitaro-vits

app.py

@@ -54,7 +54,8 @@ AVAILABLE_MODELS = {
     'KOTOBA-SPEECH-BRIA': 'kotoba-speech-bria',
     'KOTOBA-SPEECH-ALEX': 'kotoba-speech-alex',
     'KOTOBA-SPEECH-JACOB': 'kotoba-speech-jacob',
-    'BLANE-TTS': 'blane-tts'
+    'BLANE-TTS': 'blane-tts',
+    'AMITARO-VITS': 'amitaro-vits'
 }
 
 SPACE_ID = os.getenv('SPACE_ID')
@@ -243,7 +244,7 @@ if not os.path.isfile(DB_PATH):
         print("Error while downloading DB:", e)
 
 # Create DB table (if doesn't exist)
-create_db_if_missing()
+# create_db_if_missing()
     
 # Sync local DB with remote repo every 5 minute (only if a change is detected)
 scheduler = CommitScheduler(
@@ -390,7 +391,8 @@ model_names = {
     'kotoba-speech-bria': 'KOTOBA-SPEECH-v0.1-BRIA',
     'kotoba-speech-alex': 'KOTOBA-SPEECH-v0.1-ALEX',
     'kotoba-speech-jacob': 'KOTOBA-SPEECH-v0.1-JACOB',
-    'blane-tts': 'BLANE-TTS'
+    'blane-tts': 'BLANE-TTS',
+    'amitaro-vits': 'AMITARO-VITS'
     # 'styletts2': 'StyleTTS 2',
 }
 model_licenses = {
@@ -445,7 +447,8 @@ model_links = {
     'kotoba-speech-bria': 'https://kotoba-tech-kotoba-speech.hf.space/',
     'kotoba-speech-alex': 'https://kotoba-tech-kotoba-speech.hf.space/',
     'kotoba-speech-jacob': 'https://kotoba-tech-kotoba-speech.hf.space/',
-    'blane-tts': 'https://blane187-blane-tts.hf.space/'
+    'blane-tts': 'https://blane187-blane-tts.hf.space/',
+    'amitaro-vits': 'https://lycoris53-vits-tts-japanese-only-amitaro.hf.space/'
 }
 model_kwargs = {
     'moe-vits': {
@@ -454,6 +457,9 @@ model_kwargs = {
     'bark': {
         'fn_index': 3
     },
+    'amitaro-vits': {
+        'fn_index': 0
+    }
 }
 # def get_random_split(existing_split=None):
 #     choice = random.choice(list(audio_dataset.keys()))
@@ -724,6 +730,9 @@ def synthandreturn(text):
                         text,
                         f'Speaker {random.choice(range(10))} (ja)',
                     ), 
+                    'amitaro-vits': (
+                        text, 
+                    )
                 }
                 # result = router.predict(text, AVAILABLE_MODELS[model].lower(), api_name="/synthesize")
                 if model in model_kwargs:
@@ -974,5 +983,5 @@ with gr.Blocks(theme=theme, css="footer {visibility: hidden}textbox{resize:none}
             with gr.Accordion("Citation", open=False):
                 gr.Markdown(f"If you use this data in your publication, please cite us!\n\nCopy the BibTeX citation to cite this source:\n\n```bibtext\n{CITATION_TEXT}\n```\n\nPlease remember that all generated audio clips should be assumed unsuitable for redistribution or commercial use.")
 
-demo.launch()
-# demo.queue(api_open=False, default_concurrency_limit=40).launch(show_api=False)
+# demo.launch()
+demo.queue(api_open=False, default_concurrency_limit=40).launch(show_api=False)