archivartaunik commited on
Commit
df2963e
·
verified ·
1 Parent(s): 12df69b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -5
app.py CHANGED
@@ -1,14 +1,25 @@
 
1
  import os
2
  import shutil
3
- # Clone the repository and install requirements
4
- os.system("git clone https://github.com/hellcatmon/XTTSv2-Finetuning-for-New-Languages.git")
5
- os.system("pip install -r XTTSv2-Finetuning-for-New-Languages/requirements.txt")
6
-
7
  import torch
8
  import torchaudio
9
  import gradio as gr
10
  from tqdm import tqdm
11
  from huggingface_hub import snapshot_download
 
 
 
 
 
 
 
 
 
 
 
 
 
 
12
  from underthesea import sent_tokenize
13
  from TTS.tts.configs.xtts_config import XttsConfig
14
  from TTS.tts.models.xtts import Xtts
@@ -34,7 +45,7 @@ for root, _, files in os.walk(local_repo_path):
34
 
35
  print(f"Model files are saved in {destination_dir}.")
36
 
37
- # Initialize device
38
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
39
 
40
  # Load model
 
1
+ # Імпартуем патрэбныя модулі
2
  import os
3
  import shutil
 
 
 
 
4
  import torch
5
  import torchaudio
6
  import gradio as gr
7
  from tqdm import tqdm
8
  from huggingface_hub import snapshot_download
9
+
10
+ import sys
11
+
12
+ # Клонуем рэпазіторый, калі ён яшчэ не загружаны
13
+ if not os.path.exists("XTTSv2-Finetuning-for-New-Languages"):
14
+ os.system("git clone https://github.com/hellcatmon/XTTSv2-Finetuning-for-New-Languages.git")
15
+
16
+ # Перамяшчаем тэчку TTS у асноўную дырэкторыю
17
+ if os.path.exists("XTTSv2-Finetuning-for-New-Languages/TTS"):
18
+ os.system("mv XTTSv2-Finetuning-for-New-Languages/TTS ./")
19
+
20
+ # Дадаем тэчку TTS у PYTHONPATH
21
+ sys.path.append("./TTS")
22
+
23
  from underthesea import sent_tokenize
24
  from TTS.tts.configs.xtts_config import XttsConfig
25
  from TTS.tts.models.xtts import Xtts
 
45
 
46
  print(f"Model files are saved in {destination_dir}.")
47
 
48
+ # Вызначэнне прылады (выкарыстоўваецца GPU, калі даступна)
49
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
50
 
51
  # Load model