wasmdashai commited on
Commit
ee11590
1 Parent(s): f1aa1e4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ token=os.environ.get("key_")
11
  genai.configure(api_key=api_key)
12
  tokenizer = AutoTokenizer.from_pretrained("wasmdashai/vits-ar-sa-huba",token=token)
13
  #device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
14
- model_vits=VitsModel.from_pretrained("wasmdashai/vits-ar-sa-huba",token=token)#.to(device)
15
  model_vits.decoder.apply_weight_norm()
16
  # torch.nn.utils.weight_norm(self.decoder.conv_pre)
17
  # torch.nn.utils.weight_norm(self.decoder.conv_post)
 
11
  genai.configure(api_key=api_key)
12
  tokenizer = AutoTokenizer.from_pretrained("wasmdashai/vits-ar-sa-huba",token=token)
13
  #device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
14
+ model_vits=VitsModel.from_pretrained("wasmdashai/vits-ar-sa-huba-v2",token=token)#.to(device)
15
  model_vits.decoder.apply_weight_norm()
16
  # torch.nn.utils.weight_norm(self.decoder.conv_pre)
17
  # torch.nn.utils.weight_norm(self.decoder.conv_post)