Update app.py

app.py

@@ -6,7 +6,7 @@ import torchaudio
 import gradio as gr
 from tqdm import tqdm
 from huggingface_hub import snapshot_download
-
+import spaces
 import sys
 
 # Клонуем рэпазіторый, калі ён яшчэ не загружаны
@@ -23,7 +23,21 @@ sys.path.append("./TTS")
 from underthesea import sent_tokenize
 from TTS.tts.configs.xtts_config import XttsConfig
 from TTS.tts.models.xtts import Xtts
-# Define repo and model paths
+
+
+# Clone the repository if not already present
+if not os.path.exists("XTTSv2-Finetuning-for-New-Languages"):
+    os.system("git clone https://github.com/hellcatmon/XTTSv2-Finetuning-for-New-Languages.git")
+
+# Move the TTS directory to the current working directory
+if os.path.exists("XTTSv2-Finetuning-for-New-Languages/TTS"):
+    os.system("mv XTTSv2-Finetuning-for-New-Languages/TTS ./")
+
+# Add the TTS directory to the Python path
+import sys
+sys.path.append("./TTS")
+
+# Define repository and model paths
 repo_id = "archivartaunik/BE_XTTS_V2_60epoch3Dataset"
 destination_dir = "checkpoints/XTTS_v2.0_original_model_files/"
 
@@ -45,9 +59,6 @@ for root, _, files in os.walk(local_repo_path):
 
 print(f"Model files are saved in {destination_dir}.")
 
-# Вызначэнне прылады (выкарыстоўваецца GPU, калі даступна)
-device = "cuda:0" if torch.cuda.is_available() else "cpu"
-
 # Load model
 print("Loading the model...")
 xtts_checkpoint = os.path.join(destination_dir, "model.pth")
@@ -58,10 +69,10 @@ config = XttsConfig()
 config.load_json(xtts_config)
 XTTS_MODEL = Xtts.init_from_config(config)
 XTTS_MODEL.load_checkpoint(config, checkpoint_path=xtts_checkpoint, vocab_path=xtts_vocab, use_deepspeed=False)
-XTTS_MODEL.to(device)
 print("Model loaded successfully!")
 
 # Function for inference
+@spaces.GPU
 def tts_inference(belarusian_text):
     lang = "be"
     speaker_audio_file = os.path.join(destination_dir, "voice.wav")
@@ -118,4 +129,4 @@ def gradio_app():
 # Launch the app
 if __name__ == "__main__":
     app = gradio_app()
-    app.launch()
+    app.launch()