Update app.py

app.py

@@ -19,8 +19,9 @@ def get_or_load_model():
     if MODEL is None:
         print("Model not loaded, initializing...")
         try:
-            # Load the model directly to the specified DEVICE (CPU)
-            MODEL = ChatterboxTTS.from_pretrained(DEVICE)
+            # Correctly load the model by passing 'device' as a keyword argument.
+            # This ensures the model's tensors are mapped to the CPU during loading.
+            MODEL = ChatterboxTTS.from_pretrained(device=DEVICE)
             if hasattr(MODEL, 'to') and str(getattr(MODEL, 'device', 'cpu')) != DEVICE:
                 MODEL.to(DEVICE)
             print(f"Model loaded successfully. Internal device: {getattr(MODEL, 'device', 'N/A')}")