Update app.py

app.py

@@ -28,7 +28,7 @@ import os
 from VitsModelSplit.vits_model2 import VitsModel,get_state_grad_loss
 from VitsModelSplit.PosteriorDecoderModel import PosteriorDecoderModel
 from VitsModelSplit.feature_extraction import VitsFeatureExtractor
-
+from VitsModelSplit.vits_models_only_decoder imports Vits_models_only_decoder 
 from transformers import AutoTokenizer, HfArgumentParser, set_seed
 from VitsModelSplit.Arguments import DataTrainingArguments, ModelArguments, VITSTrainingArguments
 from  VitsModelSplit.dataset_features_collector import FeaturesCollectionDataset