Update app.py

app.py

@@ -128,8 +128,8 @@ def main():
                 except Exception as e:
                     st.error(f"An error occurred: {e}")
                     
-            elif selected_model == "stabilityai/stable-diffusion-xl-base-1.0" or user_input == '/image:':
-                pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, use_safetensors=True, variant="fp16")
+            elif selected_model == "stabilityai/stable-diffusion-xl-base-1.0":
+                pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float32, use_safetensors=True, variant="fp16")
                 pipe.to("cpu")
 
                 params = {'num_inference_steps': 100, 'num_images_per_prompt': 2}