Update app.py

app.py

@@ -11,7 +11,12 @@ genai.configure(api_key=api_key)
 tokenizer = AutoTokenizer.from_pretrained("asg2024/vits-ar-sa-huba",token=token)
 #device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model_vits=VitsModel.from_pretrained("asg2024/vits-ar-sa-huba",token=token)#.to(device)
-
+model_vits.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 model_vits.flow.flows:
+        torch.nn.utils.weight_norm(flow.conv_pre)
+        torch.nn.utils.weight_norm(flow.conv_post)
 
 generation_config = {
   "temperature": 1,