Update app.py

app.py

@@ -116,7 +116,7 @@ def  get_model(name_model):
 
         
         return models[name_model],tokenizer
-    models[name_model]=VitsModel.from_pretrained(name_model,token=token)
+    models[name_model]=VitsModel.from_pretrained(name_model,token=token).cuda()