Change model and employ Gradio Blocks for better app layout

app.py

@@ -9,7 +9,7 @@ import gradio as gr
 hf_api_key = os.environ['HF_API_KEY']
 
 # Text-to-image endpoint
-get_completion = pipeline("ner", model="rupeshs/LCM-runwayml-stable-diffusion-v1-5")
+get_completion = pipeline("ner", model="stable-diffusion-v1-5/stable-diffusion-v1-5")
 
 
 # A helper function to convert the PIL image to base64,
@@ -25,18 +25,27 @@ def generate(prompt):
     result_image = base64_to_pil(output)
     return result_image
 
-demo = gr.Interface(fn=generate,
-                    inputs=[gr.Textbox(label="Your prompt")],
-                    outputs=[gr.Image(label="Result")],
-                    title="Image Generation with Stable Diffusion",
-                    description="Generate any image with Stable Diffusion",
-                    allow_flagging="never",
-                    examples=[
-                        "the spirit of a tamagotchi wandering in the city of Vienna",
-                        "a mecha robot in a favela"
-                    ])
+
+with gr.Blocks() as demo:
+    gr.Markdown("# Image Generation with Stable Diffusion")
+    prompt = gr.Textbox(label="Your prompt")
+
+    with gr.Row():
+        with gr.Column():
+            negative_prompt = gr.Textbox(label="Negative prompt")
+            steps = gr.Slider(label="Inference Steps", minimum=1, maximum=100, value=25,
+                      info="In many steps will the denoiser denoise the image?")
+            guidance = gr.Slider(label="Guidance Scale", minimum=1, maximum=20, value=7,
+                      info="Controls how much the text prompt influences the result")
+            width = gr.Slider(label="Width", minimum=64, maximum=512, step=64, value=512)
+            height = gr.Slider(label="Height", minimum=64, maximum=512, step=64, value=512)
+            btn = gr.Button("Submit")
+        with gr.Column():
+            output = gr.Image(label="Result")
+
+    btn.click(fn=generate, inputs=[prompt,negative_prompt,steps,guidance,width,height], outputs=[output])
 
 demo.launch(
     share=True,
-    #server_port=int(os.environ['PORT1'])
+    #server_port=int(os.environ['PORT3'])
 )