Update app.py

app.py

@@ -21,7 +21,7 @@ import torch
 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_only_d import  Vits_models_only_decoder
 #from VitsModelSplit.vits_model import VitsModel
 from VitsModelSplit.PosteriorDecoderModel import PosteriorDecoderModel
 from VitsModelSplit.feature_extraction import VitsFeatureExtractor