Update app.py

app.py

@@ -20,27 +20,31 @@ def generate(prompt,steps,guidance,width,height):
 pipeline = DiffusionPipeline.from_pretrained("stable-diffusion-v1-5/stable-diffusion-v1-5")
 
 
-with gr.Blocks() as demo:
-    gr.Markdown("# Image Generation Demo & Test App by Srinivas")
-    gr.Markdown("## Generates an Image based on Your Promt inputted and Optional parameters selected")
-    with gr.Row():
-        with gr.Column(scale=4):
-            prompt = gr.Textbox(label="Your Prompt") #Give prompt some real estate
-        with gr.Column(scale=1, min_width=50):
-            btn = gr.Button("Submit") #Submit button side by side!
-    with gr.Accordion("Advanced options", open=False): #Let's hide the advanced options!
-            # negative_prompt = gr.Textbox(label="Negative prompt")
-            with gr.Row():
-                with gr.Column():
-                    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.0, maximum=20.0, value=7.0,
-                      info="Controls how much the text prompt influences the result")
-                with gr.Column():
-                    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)
-    output = gr.Image(label="Result") #Move the output up too
+# with gr.Blocks() as demo:
+#     gr.Markdown("# Image Generation Demo & Test App by Srinivas")
+#     gr.Markdown("## Generates an Image based on Your Promt inputted and Optional parameters selected")
+#     with gr.Row():
+#         with gr.Column(scale=4):
+#             prompt = gr.Textbox(label="Your Prompt") #Give prompt some real estate
+#         with gr.Column(scale=1, min_width=50):
+#             btn = gr.Button("Submit") #Submit button side by side!
+#     with gr.Accordion("Advanced options", open=False): #Let's hide the advanced options!
+#             # negative_prompt = gr.Textbox(label="Negative prompt")
+#             with gr.Row():
+#                 with gr.Column():
+#                     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.0, maximum=20.0, value=7.0,
+#                       info="Controls how much the text prompt influences the result")
+#                 with gr.Column():
+#                     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)
+#     output = gr.Image(label="Result") #Move the output up too
 
-    btn.click(fn=generate, inputs=[prompt,steps,guidance,width,height], outputs=[output])
+#     btn.click(fn=generate, inputs=[prompt,steps,guidance,width,height], outputs=[output])
 
-demo.launch()
+# demo.launch()
+
+from transformers import launch_gradio_demo
+
+launch_gradio_demo(generate)