Update app.py

app.py

@@ -17,6 +17,15 @@ from text.symbols import symbols
 
 limitation = os.getenv("SYSTEM") == "spaces"  # limit text and audio length in huggingface spaces
 
+if "use_mel_posterior_encoder" in hps.model.keys() and hps.model.use_mel_posterior_encoder == True:
+    print("Using mel posterior encoder for VITS2")
+    posterior_channels = 80  # vits2
+    hps.data.use_mel_posterior_encoder = True
+else:
+    print("Using lin posterior encoder for VITS1")
+    posterior_channels = hps.data.filter_length // 2 + 1
+    hps.data.use_mel_posterior_encoder = False
+
 device = 'cpu'
 
 def get_text(text, hps):