Update app.py

app.py

@@ -157,12 +157,20 @@ class VibeVoiceDemo:
     def load_model(self):
         """Load the VibeVoice model and processor."""
         print(f"Loading processor & model from {self.model_path}")
-        self.processor = VibeVoiceProcessor.from_pretrained(self.model_path)
-        self.model = VibeVoiceForConditionalGenerationInference.from_pretrained(
-            self.model_path,
-            torch_dtype=torch.bfloat16,
-            device_map=self.device,
-        )
+        self.processor = VibeVoiceProcessor.from_pretrained(self.model_path) 
+        if self.device == "cuda":
+            self.model = VibeVoiceForConditionalGenerationInference.from_pretrained(
+                self.model_path,
+                torch_dtype=torch.bfloat16,
+                device_map=self.device,
+            )
+        else:
+            self.model = VibeVoiceForConditionalGenerationInference.from_pretrained(
+                self.model_path,
+                torch_dtype=torch.float32,  # Use float32 for CPU
+                device_map="cpu",
+            )
+
         self.model.eval()
         self.model.model.noise_scheduler = self.model.model.noise_scheduler.from_config(
             self.model.model.noise_scheduler.config,