Update app.py

app.py

@@ -1,4 +1,6 @@
 import os
+import sys
+import subprocess
 import gradio as gr
 import torch
 import soundfile as sf
@@ -7,6 +9,21 @@ import librosa
 from TTS.api import TTS
 from TTS.utils.manage import ModelManager
 
+def install_sentencepiece():
+    try:
+        # Attempting to install sentencepiece via pip
+        subprocess.check_call([sys.executable, "-m", "pip", "install", "sentencepiece"])
+    except subprocess.CalledProcessError:
+        # Attempt to install sentencepiece via system package manager if pip install fails
+        if os.name == "posix":
+            os.system("sudo apt-get install -y libprotobuf10 protobuf-compiler libprotobuf-dev")
+            os.system("sudo apt-get install -y libsentencepiece-dev")
+        else:
+            raise OSError("Automatic installation of SentencePiece is not supported on this OS")
+
+# Call the function to attempt installing SentencePiece
+install_sentencepiece()
+
 # Agreeing to Coqui TTS terms of service and setting up environment variables
 os.environ["COQUI_TOS_AGREED"] = "1"
 os.makedirs("output_audio", exist_ok=True)