Update app.py

app.py

@@ -5,7 +5,7 @@ import gradio as gr
 
 pipe = StableDiffusionXLPipeline.from_pretrained(
     "segmind/SSD-1B",
-    torch_dtype = torch.float16,
+    torch_dtype = torch.float32,
     use_safetensors=True,
     # for cuda -- variant = "fp16",
 )