Update app.py

app.py

@@ -75,7 +75,7 @@ def get_model(name_model):
         tokenizer = AutoTokenizer.from_pretrained(name_model, token=token)
         return models[name_model], tokenizer
 
-    models[name_model] = VitsModel.from_pretrained(name_model, token=token).cuda()
+    models[name_model] = VitsModel.from_pretrained(name_model, token=token) 
     models[name_model].decoder.apply_weight_norm()
     for flow in models[name_model].flow.flows:
         torch.nn.utils.weight_norm(flow.conv_pre)