Update app.py

app.py

@@ -1,8 +1,10 @@
 import gradio as gr
 GK=0
-from transformers import AutoTokenizer,VitsModel
+from transformers import AutoTokenizer
 import torch
 import os
+from VitsModelSplit.vits_model2 import VitsModel,get_state_grad_loss
+
 token=os.environ.get("key_")
 tokenizer = AutoTokenizer.from_pretrained("wasmdashai/vits-ar-sa-huba",token=token)
 #device = torch.device("cuda" if torch.cuda.is_available() else "cpu")