Upload app.py

app.py

@@ -233,7 +233,7 @@ SAMPLE_RATE = 24000
 
 @torch.no_grad()
 def forward(tokens, voices, speed, sk, device='cpu'):
-    assert sk == os.environ['SK'], sk
+    assert sk in {os.environ['SK'], os.environ['ARENA']}, sk
     ref_s = torch.mean(torch.stack([VOICES[device][v][len(tokens)] for v in voices]), dim=0)
     tokens = torch.LongTensor([[0, *tokens, 0]]).to(device)
     input_lengths = torch.LongTensor([tokens.shape[-1]]).to(device)
@@ -292,7 +292,7 @@ def generate(text, voice='af', ps=None, speed=1, trim=0.5, use_gpu='auto', sk=No
     ps = ps or phonemize(text, voice)
     if sk not in {os.environ['SK'], os.environ['ARENA']}:
         assert text in sents or ps.strip('"') in harvsents, ('❌', datetime.now(), text, voice, use_gpu, sk)
-        sk = os.environ['SK']
+        sk = os.environ['ARENA']
     voices = resolve_voices(voice, warn=ps)
     speed = clamp_speed(speed)
     trim = clamp_trim(trim)