Fix syntax error in TTS stage and complete pipeline

app.py

@@ -307,7 +307,8 @@ class TransformerTTS(nn.Module):
 
     @torch.no_grad()
     def inference(self, text, max_length=800, gate_threshold=1e-5, with_tqdm=True):
-
+        self.eval()
+        self.train(False)
         text_lengths = torch.tensor(text.shape[1]).unsqueeze(0).to(DEVICE)
         N = 1
         SOS = torch.zeros((N, 1, hp.mel_freq), device=DEVICE)
@@ -379,18 +380,7 @@ try:
         TTS_MODEL.load_state_dict(state["state_dict"])
     else:
         TTS_MODEL.load_state_dict(state) # Assume the whole file is the state_dict
-
     TTS_MODEL.eval()
-
-    # Set all submodules to eval mode and disable gradients permanently
-    for module in TTS_MODEL.modules():
-        if hasattr(module, 'training'):
-            module.train(False)
-    
-    # Disable gradients for all parameters permanently
-    for param in TTS_MODEL.parameters():
-        param.requires_grad = False
-
     
     # Try torch.compile for additional speedup (PyTorch 2.0+)
     try: