Update app.py

app.py

@@ -22,7 +22,7 @@ from datasets import DatasetDict,Dataset
 import os
 from VitsModelSplit.vits_model2 import VitsModel,get_state_grad_loss
 from VitsModelSplit.vits_model_only_d import  Vits_models_only_decoder
-from VitsModelSplit.vits_model import VitsModel
+#from VitsModelSplit.vits_model import VitsModel
 from VitsModelSplit.PosteriorDecoderModel import PosteriorDecoderModel
 from VitsModelSplit.feature_extraction import VitsFeatureExtractor
 from transformers import AutoTokenizer, HfArgumentParser, set_seed