wasmdashai commited on
Commit
ef9e856
·
verified ·
1 Parent(s): b553e52

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -7
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
- %cd VitsModelSplit # If you restart runtime
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))