Upload app.py

app.py

@@ -34,18 +34,15 @@ def load_models_once():
     try:
         print("🧠 Loading Llasa-3B...")
         
-        # Add paths for local modules
+        # Import required libraries
         import sys
-        sys.path.append('./Llasa-3B')
-        sys.path.append('./xcodec2')
         
         from transformers import AutoTokenizer, AutoModelForCausalLM
         
-        # Load Llasa-3B from local directory
-        tokenizer = AutoTokenizer.from_pretrained("./Llasa-3B", local_files_only=True)
+        # Load Llasa-3B from Hugging Face Hub
+        tokenizer = AutoTokenizer.from_pretrained("HKUSTAudio/Llasa-3B")
         model = AutoModelForCausalLM.from_pretrained(
-            "./Llasa-3B",
-            local_files_only=True,
+            "HKUSTAudio/Llasa-3B",
             torch_dtype=torch.float16 if device.type != 'cpu' else torch.float32,
             low_cpu_mem_usage=True
         )
@@ -57,9 +54,9 @@ def load_models_once():
         print("✅ Llasa-3B loaded successfully!")
         
         print("🎵 Loading XCodec2...")
-        from modeling_xcodec2 import XCodec2Model
+        from xcodec2.modeling_xcodec2 import XCodec2Model
         
-        codec_model = XCodec2Model.from_pretrained("./xcodec2", local_files_only=True)
+        codec_model = XCodec2Model.from_pretrained("HKUSTAudio/xcodec2")
         
         if device.type != 'cpu':
             try: