Update app.py

app.py

@@ -6,13 +6,13 @@ from diffusers import StableDiffusionXLPipeline
 pipe = StableDiffusionXLPipeline.from_pretrained(
     "stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16
 )
-pipe = pipe.to("cuda")  # Move the model to GPU for faster processing
+pipe = pipe.to("cpu")  # Move the model to GPU for faster processing
 
 # Create a Streamlit app
-st.title("Image Generation App")
+st.title("SDXL Turbo")
 
 # Add a text input for the prompt
-prompt = st.text_input("Prompt", "a photo of an astronaut riding a horse on mars")
+prompt = st.text_input("Prompt")
 
 # Generate the image based on the prompt
 if st.button("Generate Image"):