Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -4,12 +4,7 @@ from transformers import AutoTokenizer
|
|
4 |
import torch
|
5 |
import os
|
6 |
from VitsModelSplit.vits_model2 import VitsModel,get_state_grad_loss
|
7 |
-
|
8 |
-
%cd monotonic_align
|
9 |
-
|
10 |
-
!python setup.py build_ext --inplace
|
11 |
-
%cd ..
|
12 |
-
%cd ..
|
13 |
|
14 |
token=os.environ.get("key_")
|
15 |
tokenizer = AutoTokenizer.from_pretrained("wasmdashai/vits-ar-sa-huba",token=token)
|
@@ -46,7 +41,7 @@ model=VitsModel.from_pretrained("facebook/mms-tts-eng").to(device)
|
|
46 |
# model.discriminator=model1.discriminator
|
47 |
# model.duration_predictor=model1.duration_predictor
|
48 |
|
49 |
-
model.setMfA(monotonic_align.maximum_path)
|
50 |
# tokenizer = AutoTokenizer.from_pretrained("facebook/mms-tts-ara",cache_dir="./")
|
51 |
feature_extractor = VitsFeatureExtractor()
|
52 |
parser = HfArgumentParser((ModelArguments, DataTrainingArguments, VITSTrainingArguments))
|
|
|
4 |
import torch
|
5 |
import os
|
6 |
from VitsModelSplit.vits_model2 import VitsModel,get_state_grad_loss
|
7 |
+
|
|
|
|
|
|
|
|
|
|
|
8 |
|
9 |
token=os.environ.get("key_")
|
10 |
tokenizer = AutoTokenizer.from_pretrained("wasmdashai/vits-ar-sa-huba",token=token)
|
|
|
41 |
# model.discriminator=model1.discriminator
|
42 |
# model.duration_predictor=model1.duration_predictor
|
43 |
|
44 |
+
# model.setMfA(monotonic_align.maximum_path)
|
45 |
# tokenizer = AutoTokenizer.from_pretrained("facebook/mms-tts-ara",cache_dir="./")
|
46 |
feature_extractor = VitsFeatureExtractor()
|
47 |
parser = HfArgumentParser((ModelArguments, DataTrainingArguments, VITSTrainingArguments))
|