Update app.py

app.py

@@ -113,9 +113,9 @@ def generate_speech(text: str, speaker_name: str, cfg_scale: float = 1.5, progre
         inputs = PROCESSOR.process_input_with_cached_prompt(
             text=text.strip(), cached_prompt=all_prefilled_outputs, padding=True, return_tensors="pt", return_attention_mask=True
         )
-        MODEL.to("cuda")
+        MODEL.to("cpu")
         for k, v in inputs.items():
-            if torch.is_tensor(v): inputs[k] = v.to("cuda")
+            if torch.is_tensor(v): inputs[k] = v.to("cpu")
 
         progress(0.4, desc="Generating speech...")
         start_time = time.time()