Update app.py

app.py

@@ -28,8 +28,8 @@ gpt2_model = GPT2LMHeadModel.from_pretrained("gpt2").to(device)
 dtype = torch.float16 if torch.cuda.is_available() else torch.float32
 pipe = StableDiffusionPipeline.from_pretrained(
     "runwayml/stable-diffusion-v1-5",
-    torch_dtype=dtype
-)
+    torch_dtype=torch.float32  # Required for CPU
+).to("cpu")
 pipe = pipe.to(device)
 
 # Emotion detection for Text-to-Audio