Update app.py

app.py

@@ -4,7 +4,8 @@ import gradio as gr
 from diffusers import StableDiffusionImg2ImgPipeline
 
 # Load the Stable Diffusion img2img pipeline
-device = "cpu"
+device = "cuda" if torch.cuda.is_available() else "cpu"
+
 pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
     "runwayml/stable-diffusion-v1-5",
     torch_dtype=torch.float16,
@@ -37,7 +38,7 @@ def generate_image(init_image, prompt_choice, strength, guidance_scale):
 
 # Gradio interface
 with gr.Blocks() as demo:
-    gr.Markdown("# 🦑 Corroded Pipeline Generator - Underwater Img2Img")
+    gr.Markdown("Corroded Pipeline Generator - Underwater Img2Img")
     with gr.Row():
         with gr.Column():
             init_image = gr.Image(label="Upload Initial Image", type="pil")