Update app.py

app.py

@@ -112,6 +112,12 @@ def  get_model(name_model):
     if name_model in   models:
         return models[name_model]
     models[name_model]=VitsModel.from_pretrained(name_model,token=api_key).to(device)
+    models[name_model].decoder.apply_weight_norm()
+    # torch.nn.utils.weight_norm(self.decoder.conv_pre)
+    # torch.nn.utils.weight_norm(self.decoder.conv_post)
+    for flow in models[name_model].flow.flows:
+        torch.nn.utils.weight_norm(flow.conv_pre)
+        torch.nn.utils.weight_norm(flow.conv_post)
     return models[name_model]
 
 def  genrate_speech(text,name_model):