main

app.py

@@ -18,37 +18,11 @@ from huggingface_hub import hf_hub_download, snapshot_download
 from underthesea import sent_tokenize
 from unidecode import unidecode
 from vinorm import TTSnorm
-
+from TTS.tts.configs.xtts_config import XttsConfig
+from TTS.tts.models.xtts import Xtts
 print("Setting up the environment...")
 import subprocess
 
-setup_command = [
-    "pip", "install", "--use-deprecated=legacy-resolver", "-e", ".", "-q",
-    "&&", "cd", "TTS ",
-    "&&", "git submodule update --init --recursive",
-    "&&", "echo", "Installing other requirements...",
-    "&&", "pip", "install", "-r", "requirements.txt", "-q",
-    "&&", "echo", "Downloading Japanese/Chinese tokenizer...",
-    "&&", "python", "-m", "unidic", "download",
-    "&&", "pip", "install", "--upgrade", "gradio"
-]
-
-try:
-    subprocess.run(setup_command, check=True, shell=True)
-    print("Environment setup completed successfully.")
-except subprocess.CalledProcessError as e:
-    print(f"Error during environment setup: {e}")
-    print("Please run the setup command manually if needed.")
-
-print("\nTo set up the environment manually, run the following command:")
-print("pip install --use-deprecated=legacy-resolver -e . -q && \\ ")
-print("cd .. && \\ ")
-print("echo \"Installing other requirements...\" && \\ ")
-print("pip install -r requirements.txt -q && \\ ")
-print("echo \"Downloading Japanese/Chinese tokenizer...\" && \\ ")
-print("python -m unidic download && \\ ")
-print("pip install --upgrade gradio")
-
 XTTS_MODEL = None
 SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
 MODEL_DIR = os.path.join(SCRIPT_DIR, "model")
@@ -84,20 +58,21 @@ def load_model(checkpoint_dir="model/", repo_id="capleaf/viXTTS", use_deepspeed=
         yield f"Model download finished..."
 
     xtts_config = os.path.join(checkpoint_dir, "config.json")
-    from TTS.tts.configs.xtts_config import XttsConfig
-    from TTS.tts.models.xtts import Xtts
+    
     config = XttsConfig()
     config.load_json(xtts_config)
     XTTS_MODEL = Xtts.init_from_config(config)
-    yield "Loading model..."
-    XTTS_MODEL.load_checkpoint(
-        config, checkpoint_dir=checkpoint_dir, use_deepspeed=use_deepspeed
-    )
-    if torch.cuda.is_available():
-        XTTS_MODEL.cuda()
-
-    print("Model Loaded!")
-    yield "Model Loaded!"
+    yield "Đang tải model.."
+    try: 
+        XTTS_MODEL.load_checkpoint(
+            config, checkpoint_dir=checkpoint_dir, use_deepspeed=use_deepspeed
+        )
+        if torch.cuda.is_available():
+            XTTS_MODEL.cuda()
+    except: 
+        print("error")
+    finally:
+        yield "Đã tải model!"
 
 
 # Define dictionaries to store cached results
@@ -362,7 +337,7 @@ if __name__ == "__main__":
                     value=True, label="Sử dụng DeepSpeed ​​để suy luận nhanh hơn"
                 )
 
-                progress_load = gr.Label(label="Trạng thái:")
+                progress_load = gr.Label(label="Tiến trình:")
                 load_btn = gr.Button(
                     value="Khởi tạo model", variant="primary"
                 )
@@ -399,7 +374,7 @@ if __name__ == "__main__":
                 )
 
                 use_filter = gr.Checkbox(
-                    label="Lọc tiếp ồn",
+                    label="Lọc tiếng ồn",
                     value=True,
                 )