Update wrapper.py

wrapper.py

@@ -4,23 +4,6 @@ import os
 import subprocess
 import sys
 
-def debug_environment():
-    """Print debug information about the environment"""
-    print("\n=== Debug Information ===")
-    print(f"Current working directory: {os.getcwd()}")
-    print("\nDirectory contents:")
-    for root, dirs, files in os.walk("."):
-        level = root.replace(".", "").count(os.sep)
-        indent = " " * 4 * level
-        print(f"{indent}{os.path.basename(root)}/")
-        subindent = " " * 4 * (level + 1)
-        for f in files:
-            print(f"{subindent}{f}")
-    
-    print("\nEnvironment variables:")
-    for key, value in os.environ.items():
-        print(f"{key}={value}")
-
 def main():
     parser = argparse.ArgumentParser(description='Run YuE model with direct input')
     parser.add_argument('--genre', type=str, required=True, help='Genre tags for the music')
@@ -33,7 +16,12 @@ def main():
     args = parser.parse_args()
 
     print("\n=== Starting YuE Inference ===")
-    debug_environment()
+    current_dir = os.path.dirname(os.path.abspath(__file__))
+    inference_dir = os.path.dirname(os.path.abspath(__file__))
+    
+    print(f"\nCurrent directory: {os.getcwd()}")
+    print(f"Script directory: {current_dir}")
+    print(f"Inference directory: {inference_dir}")
 
     # Create temporary files for genre and lyrics
     with tempfile.NamedTemporaryFile(mode='w', delete=False) as genre_file:
@@ -51,18 +39,22 @@ def main():
     output_dir = '/home/user/app/output'
     
     try:
-        # Get the directory where wrapper.py is located
-        current_dir = os.path.dirname(os.path.abspath(__file__))
-        infer_script = os.path.join(current_dir, 'infer.py')
+        # Go to inference directory for running the script
+        os.chdir(inference_dir)
+        print(f"\nChanged working directory to: {os.getcwd()}")
         
+        infer_script = 'infer.py'
         print(f"\nInference script path: {infer_script}")
         print(f"Script exists: {os.path.exists(infer_script)}")
         
+        tokenizer_path = './mm_tokenizer_v0.2_hf/tokenizer.model'
+        print(f"\nChecking tokenizer at: {tokenizer_path}")
+        print(f"Tokenizer exists: {os.path.exists(tokenizer_path)}")
         if os.path.exists('./mm_tokenizer_v0.2_hf'):
-            print("\nTokenizer directory contents:")
+            print("Tokenizer directory contents:")
             print(os.listdir('./mm_tokenizer_v0.2_hf'))
         else:
-            print("\nWARNING: Tokenizer directory not found!")
+            print("WARNING: Tokenizer directory not found!")
             
         print("\nExecuting inference command...")
         command = [
@@ -80,7 +72,6 @@ def main():
         
         print(f"Command: {' '.join(command)}")
         
-        # Run the inference script
         result = subprocess.run(command, 
                               check=True,
                               capture_output=True,