Update app.py

app.py

@@ -7,10 +7,10 @@ from PIL import Image
 
 app = FastAPI()
 
-# Safe GPU detection
-device = "cuda" if torch.cuda.is_available() else "cpu"
-dtype = torch.float16 if device == "cuda" else torch.float32
-variant = "fp16" if device == "cuda" else None
+# Force CPU setup
+device = "cpu"
+dtype = torch.float32
+variant = None  # Don't use "fp16" on CPU
 
 # Load text-to-image model
 text2img_pipe = StableDiffusionPipeline.from_pretrained(