Update app.py

app.py

@@ -3,19 +3,26 @@ from diffusers import StableDiffusionPipeline
 import torch
 
 model_id = "runwayml/stable-diffusion-v1-5"
-pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32)
+pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32, safety_checker=None)
 
 
 def infer(prompt):
 
     prompt = "a photo of an astronaut riding a horse on mars"
-    image = pipe(prompt=prompt, guidance_scale=10.0, num_inference_steps=12, width=256, height=256).images[0]  
+    image = pipe(prompt=prompt, guidance_scale=10.0, num_inference_steps=6, width=256, height=256).images[0]  
 
     return image
 
-with gr.Blocks() as demo:
+css="""
+#col-container {
+    margin: 0 auto;
+    max-width: 720px;
+}
+"""
+
+with gr.Blocks(css=css) as demo:
     
-    with gr.Column():
+    with gr.Column(elem_id="col-container"):
         
         with gr.Row():
             prompt = gr.Text(