Upload app.py

app.py

@@ -233,7 +233,7 @@ def clamp_speed(speed):
     return speed
 
 # Must be backwards compatible with https://huggingface.co/spaces/Pendrokar/TTS-Spaces-Arena
-def generate(text, voice='af', ps=None, speed=1, trim=3000, use_gpu='auto'):
+def generate(text, voice='af', ps=None, speed=1, trim=3000, use_gpu='auto', sh=None):
     voices = resolve_voices(voice, warn=ps)
     ps = ps or phonemize(text, voice)
     speed = clamp_speed(speed)
@@ -246,9 +246,7 @@ def generate(text, voice='af', ps=None, speed=1, trim=3000, use_gpu='auto'):
         tokens = tokens[:510]
     ps = ''.join(next(k for k, v in VOCAB.items() if i == v) for i in tokens)
     use_gpu = len(ps) > 99 if use_gpu == 'auto' else use_gpu
-    is_space = os.environ.get('SPACE_ID') is not None
-    is_api_call = os.environ.get('GRADIO_ENDPOINT') is not None
-    print('🔥', datetime.now(), text, voices, ps, use_gpu, is_space, is_api_call)
+    print('🔥', datetime.now(), text, voices, ps, use_gpu, is_space, sh)
     try:
         if use_gpu:
             out = forward_gpu(tokens, voices, speed)
@@ -325,6 +323,8 @@ with gr.Blocks() as basic_tts:
                         voice.change(lambda v, b: gr.Button(b, variant='primary' if v.startswith(b[:2]) else 'secondary'), inputs=[voice, btn], outputs=[btn])
     text.submit(generate, inputs=[text, voice, in_ps, speed, trim, use_gpu], outputs=[audio, out_ps])
     generate_btn.click(generate, inputs=[text, voice, in_ps, speed, trim, use_gpu], outputs=[audio, out_ps])
+    sh = gr.State()
+    basic_tts.load(lambda r: r.session_hash, None, sh)
 
 @torch.no_grad()
 def lf_forward(token_lists, voices, speed, device='cpu'):