ASG Models commited on
Commit
2a84d5f
·
verified ·
1 Parent(s): 6273feb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -15,8 +15,9 @@ tokenizer = AutoTokenizer.from_pretrained("asg2024/vits-ar-sa-huba",token=api_ke
15
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
16
  def get_model(name_model):
17
  global models
18
- if name_model in not models:
19
- models[name_model]=VitsModel.from_pretrained(name_model,token=api_key).to(device)
 
20
  return models[name_model]
21
 
22
  def genrate_speech(text,name_model):
 
15
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
16
  def get_model(name_model):
17
  global models
18
+ if name_model in models:
19
+ return models[name_model]
20
+ models[name_model]=VitsModel.from_pretrained(name_model,token=api_key).to(device)
21
  return models[name_model]
22
 
23
  def genrate_speech(text,name_model):