Upload app.py with huggingface_hub

app.py

@@ -30,27 +30,59 @@ if not vibevoice_dir.exists():
 import sys
 sys.path.insert(0, str(vibevoice_dir))
 
-# Change to VibeVoice directory for imports
-original_cwd = os.getcwd()
-os.chdir(vibevoice_dir)
-
+# Import VibeVoice modules
 try:
-    from modular.configuration_vibevoice import VibeVoiceConfig
-    from modular.modeling_vibevoice_inference import VibeVoiceForConditionalGenerationInference
-    from processor.vibevoice_processor import VibeVoiceProcessor
-    from modular.streamer import AudioStreamer
-except ImportError as e:
-    os.chdir(original_cwd)
-    raise ImportError(
-        f"VibeVoice module not found. Error: {e}\n"
-        "Please ensure VibeVoice is properly installed:\n"
-        "git clone https://github.com/microsoft/VibeVoice.git\n"
-        "cd VibeVoice/\n"
-        "pip install -e .\n"
-    )
-finally:
-    # Restore original directory after imports
-    os.chdir(original_cwd)
+    # Try direct import first (if installed as package)
+    from vibevoice.modular.configuration_vibevoice import VibeVoiceConfig
+    from vibevoice.modular.modeling_vibevoice_inference import VibeVoiceForConditionalGenerationInference
+    from vibevoice.processor.vibevoice_processor import VibeVoiceProcessor
+    from vibevoice.modular.streamer import AudioStreamer
+except ImportError:
+    try:
+        # Try importing from the cloned directory
+        import importlib.util
+        
+        # Load modules directly from the VibeVoice directory
+        def load_module(module_name, file_path):
+            spec = importlib.util.spec_from_file_location(module_name, file_path)
+            module = importlib.util.module_from_spec(spec)
+            sys.modules[module_name] = module
+            spec.loader.exec_module(module)
+            return module
+        
+        # Load each module
+        config_module = load_module(
+            "vibevoice_config",
+            vibevoice_dir / "modular" / "configuration_vibevoice.py"
+        )
+        VibeVoiceConfig = config_module.VibeVoiceConfig
+        
+        model_module = load_module(
+            "vibevoice_model",
+            vibevoice_dir / "modular" / "modeling_vibevoice_inference.py"
+        )
+        VibeVoiceForConditionalGenerationInference = model_module.VibeVoiceForConditionalGenerationInference
+        
+        processor_module = load_module(
+            "vibevoice_processor",
+            vibevoice_dir / "processor" / "vibevoice_processor.py"
+        )
+        VibeVoiceProcessor = processor_module.VibeVoiceProcessor
+        
+        streamer_module = load_module(
+            "vibevoice_streamer",
+            vibevoice_dir / "modular" / "streamer.py"
+        )
+        AudioStreamer = streamer_module.AudioStreamer
+        
+    except Exception as e:
+        raise ImportError(
+            f"VibeVoice module not found. Error: {e}\n"
+            "Please ensure VibeVoice is properly installed:\n"
+            "git clone https://github.com/microsoft/VibeVoice.git\n"
+            "cd VibeVoice/\n"
+            "pip install -e .\n"
+        )
 from transformers.utils import logging
 from transformers import set_seed