ASG Models commited on
Commit
dce3b4a
1 Parent(s): cf86894

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,8 +9,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
 
15
 
16
  generation_config = {
 
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
 
15
 
16
  generation_config = {