wasmdashai commited on
Commit
9441ab4
·
verified ·
1 Parent(s): 7c7b2f8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -1,8 +1,10 @@
1
  import gradio as gr
2
  GK=0
3
- from transformers import AutoTokenizer,VitsModel
4
  import torch
5
  import os
 
 
6
  token=os.environ.get("key_")
7
  tokenizer = AutoTokenizer.from_pretrained("wasmdashai/vits-ar-sa-huba",token=token)
8
  #device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
1
  import gradio as gr
2
  GK=0
3
+ from transformers import AutoTokenizer
4
  import torch
5
  import os
6
+ from VitsModelSplit.vits_model2 import VitsModel,get_state_grad_loss
7
+
8
  token=os.environ.get("key_")
9
  tokenizer = AutoTokenizer.from_pretrained("wasmdashai/vits-ar-sa-huba",token=token)
10
  #device = torch.device("cuda" if torch.cuda.is_available() else "cpu")