wasmdashai commited on
Commit
c39b385
·
verified ·
1 Parent(s): 2d7efb5

Update app.py

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