Update app.py

app.py

@@ -21,6 +21,8 @@ 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 import VitsModel
 from VitsModelSplit.PosteriorDecoderModel import PosteriorDecoderModel
 from VitsModelSplit.feature_extraction import VitsFeatureExtractor
 from transformers import AutoTokenizer, HfArgumentParser, set_seed