Create app.py

app.py

@@ -0,0 +1,42 @@
+import gradio as gr
+import torch
+from TTS.tts.configs.xtts_config import XttsConfig
+from TTS.tts.models.xtts import Xtts
+from huggingface_hub import hf_hub_download
+
+# Вызначэнне прылады (выкарыстоўваецца GPU, калі даступна)
+device = "cuda:0" if torch.cuda.is_available() else "cpu"
+
+# Шлях да мадэлі ў Hugging Face
+repo_id = "archivartaunik/GPT_XTTS_V2_40EP"  # Імя рэпазіторыя Hugging Face
+checkpoint_file = hf_hub_download(repo_id, filename="model.pth")  # Загружаем файл праверачнай кропкі
+config_file = hf_hub_download(repo_id, filename="config.json")    # Загружаем канфігурацыю
+vocab_file = hf_hub_download(repo_id, filename="vocab.json")      # Загружаем слоўнік
+
+# Загрузка канфігурацыі мадэлі
+config = XttsConfig()
+config.load_json(config_file)
+
+# Ініцыялізацыя і загрузка мадэлі
+XTTS_MODEL = Xtts.init_from_config(config)
+XTTS_MODEL.load_checkpoint(config, checkpoint_path=checkpoint_file, vocab_path=vocab_file, use_deepspeed=False)
+XTTS_MODEL.to(device)
+
+# Функцыя для пераўтварэння тэксту ў гукавы фармат
+def text_to_speech(input_text):
+    # Генерацыя аўдыя на аснове ўваходнага тэксту
+    output = XTTS_MODEL.generate(input_text)
+    return output.cpu().numpy()
+
+# Налада інтэрфейсу Gradio
+demo = gr.Interface(
+    fn=text_to_speech,            # Функцыя пераўтварэння
+    inputs="text",                # Уваходныя дадзеныя (тэкст)
+    outputs="audio",              # Выхадныя дадзеныя (аўдыя)
+    title="XTTS Belarusian Demo",        # Назва праекта
+    description="Увядзіце тэкст, і мадэль пераўтворыць яго ў аўдыя.",  # Апісанне
+)
+
+# Запуск прыкладання
+if __name__ == "__main__":
+    demo.launch()