Update app.py

app.py

@@ -3,6 +3,7 @@ import torch
 from transformers import pipeline
 
 # Loading the TTS and Vocoder ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+import sys
 import re
 import json
 import inflect
@@ -45,16 +46,27 @@ if not os.path.exists('yarngpt'):
 else:
     print("YarnGPT repository already exists")
 
-# Install required packages
-print("Installing required packages...")
-run_command("pip install -q outetts uroman")
-
-# If you need to install the cloned package in development mode
-if os.path.exists('yarngpt'):
-    os.chdir('yarngpt')
-    run_command("pip install -e .")
-    os.chdir('..')  # Go back to the original directory
-    
+# # Install required packages with specific versions
+# print("Installing required packages with specific versions...")
+# run_command("pip install -q outetts==0.3.3 uroman==1.3.1.1")
+
+# Add the yarngpt directory to Python path instead of installing it
+yarngpt_path = os.path.join(os.getcwd(), 'yarngpt')
+if os.path.exists(yarngpt_path) and yarngpt_path not in sys.path:
+    sys.path.append(yarngpt_path)
+    print(f"Added {yarngpt_path} to Python path")
+
+# Now you should be able to import from yarngpt
+# Import this after adding to path
+try:
+    from yarngpt.audiotokenizer import AudioTokenizerV2
+    print("Successfully imported AudioTokenizerV2 from yarngpt")
+except ImportError as e:
+    print(f"Error importing from yarngpt: {e}")
+    # Check the content of the directory to debug
+    if os.path.exists(yarngpt_path):
+        print("Contents of yarngpt directory:")
+        print(os.listdir(yarngpt_path))
 
 # Download files using Python's requests library instead of !wget
 def download_file(url, save_path):