Update app.py

app.py

@@ -1,7 +1,7 @@
 """
 VibeVoice Gradio Demo - High-Quality Dialogue Generation Interface with Streaming Support
 """
-
+import spaces
 import argparse
 import json
 import os
@@ -43,7 +43,8 @@ class VibeVoiceDemo:
         self.load_model()
         self.setup_voice_presets()
         self.load_example_scripts()  # Load example scripts
-        
+
+    @spaces.GPU
     def load_model(self):
         """Load the VibeVoice model and processor."""
         print(f"Loading processor & model from {self.model_path}")
@@ -126,7 +127,8 @@ class VibeVoiceDemo:
         except Exception as e:
             print(f"Error reading audio {audio_path}: {e}")
             return np.array([])
-    
+
+    @spaces.GPU
     def generate_podcast_streaming(self, 
                                  num_speakers: int,
                                  script: str,
@@ -418,7 +420,8 @@ class VibeVoiceDemo:
             import traceback
             traceback.print_exc()
             yield None, None, error_msg, gr.update(visible=False)
-    
+
+    @spaces.GPU
     def _generate_with_streamer(self, inputs, cfg_scale, audio_streamer):
         """Helper method to run generation with streamer in a separate thread."""
         try:
@@ -1137,9 +1140,9 @@ def main():
     
     # Initialize demo instance
     demo_instance = VibeVoiceDemo(
-        model_path=args.model_path,
-        device=args.device,
-        inference_steps=args.inference_steps
+        model_path='microsoft/VibeVoice-1.5B',
+        device='cuda',
+        inference_steps=10
     )
     
     # Create interface
@@ -1155,11 +1158,7 @@ def main():
     try:
         interface.queue(
             max_size=20,  # Maximum queue size
-            default_concurrency_limit=1  # Process one request at a time
         ).launch(
-            share=args.share,
-            # server_port=args.port,
-            server_name="0.0.0.0" if args.share else "127.0.0.1",
             show_error=True,
             show_api=False  # Hide API docs for cleaner interface
         )