Update app.py

app.py

@@ -7,7 +7,10 @@ import gradio as gr
 # Load the Stable Diffusion model
 model_id = "runwayml/stable-diffusion-v1-5"
 pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
-pipe = pipe.to("cuda")
+
+# Use CPU as fallback if CUDA is not available
+device = "cuda" if torch.cuda.is_available() else "cpu"
+pipe = pipe.to(device)
 
 # Function to generate an image
 def generate_image(prompt):
@@ -16,8 +19,14 @@ def generate_image(prompt):
 
 # Define the Gradio interface
 with gr.Blocks() as demo:
-    gr.Markdown("# 🌄 Generate Stunning Images with Stable Diffusion")
-    gr.Markdown("### Type a prompt to create a beautiful image:")
+    gr.HTML(
+        """
+        <div style="text-align: center;">
+            <h1>🌄 Generate Stunning Images with Stable Diffusion</h1>
+            <h3>Type a prompt to create a beautiful image:</h3>
+        </div>
+        """
+    )
     
     with gr.Row():
         with gr.Column():
@@ -28,7 +37,13 @@ with gr.Blocks() as demo:
     
     submit_button.click(fn=generate_image, inputs=prompt, outputs=image_output)
     
-    gr.Markdown("#### Developed by Salman Maqbool ✨")
+    gr.HTML(
+        """
+        <div style="text-align: center;">
+            <h4>Developed by Salman Maqbool ✨</h4>
+        </div>
+        """
+    )
 
 # Launch the Gradio app
-demo.launch()
+demo.launch()