Update app.py

app.py

@@ -18,7 +18,8 @@ tts = TTS(model_name="tts_models/en/ljspeech/tacotron2-DDC")
 # Initialize Stable Diffusion pipeline
 pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", torch_dtype=torch.float16)
 pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
-pipe = pipe.to("cuda")
+pipe = pipe.to("cpu")
+
 
 def generate_text(prompt, max_length=200):
     input_ids = tokenizer.encode(prompt, return_tensors="pt")