Steveeeeeeen HF Staff commited on
Commit
3ccd822
·
verified ·
1 Parent(s): 3d52831

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -68,7 +68,7 @@ class VibeVoiceDemo:
68
  print(f"Error reading audio {audio_path}: {e}")
69
  return np.array([])
70
 
71
- @GPU(duration=60)
72
  def generate_podcast(self,
73
  num_speakers: int,
74
  script: str,
@@ -413,7 +413,7 @@ def create_demo_interface(demo_instance: VibeVoiceDemo):
413
 
414
 
415
  def run_demo(
416
- model_path: str = "microsoft/VibeVoice-1.5B",
417
  device: str = "cuda",
418
  inference_steps: int = 5,
419
  share: bool = True,
 
68
  print(f"Error reading audio {audio_path}: {e}")
69
  return np.array([])
70
 
71
+ @GPU(duration=180)
72
  def generate_podcast(self,
73
  num_speakers: int,
74
  script: str,
 
413
 
414
 
415
  def run_demo(
416
+ model_path: str = "microsoft/VibeVoice-Large",
417
  device: str = "cuda",
418
  inference_steps: int = 5,
419
  share: bool = True,