Rename gradio_demo.py to app.py

gradio_demo.py → app.py

@@ -19,6 +19,7 @@ import soundfile as sf
 import torch
 import os
 import traceback
+import spaces
 
 from vibevoice.modular.configuration_vibevoice import VibeVoiceConfig
 from vibevoice.modular.modeling_vibevoice_inference import VibeVoiceForConditionalGenerationInference
@@ -126,7 +127,7 @@ 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,
@@ -423,6 +424,7 @@ class VibeVoiceDemo:
             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: