Update app.py

app.py

@@ -8,6 +8,7 @@ from diffusers import StableDiffusionPipeline
 from diffusers import OnnxRuntimeModel
 import torch
 
+black app.py
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model_repo_id = "runwayml/stable-diffusion-v1-5"  # Replace to the model you would like to use
@@ -109,13 +110,22 @@ css = """
 
 # Interface Gradio
         with gr.Blocks() as demo:
-             gr.Markdown("## Text-to-Image Optimized for CPU")
+             with gr.Column(elem_id="col-container"):
+            gr.Markdown("## Text-to-Image Optimized for CPU")
         
-        with gr.Row():
-            prompt = gr.Textbox(label="Prompt")
-            num_inference_steps = gr.Slider(
-            label="Inference Steps", minimum=1, maximum=50, step=1, value=15
-        )
+            with gr.Row():
+               prompt = gr.Textbox(
+                   label="Prompt",
+                   show_label=False,
+                   max_lines=1,
+                   placeholder="Enter your prompt",
+                   container=False,
+            )
+            run_button = gr.Button("Run", scale=0, variant="primary")
+
+        result = gr.Image(label="Result", show_label=False)
+
+
         with gr.Row():
              generate_button = gr.Button("Generate")
              result = gr.Image(label="Generated Image")