Update app.py

app.py

@@ -39,6 +39,57 @@ with gr.Blocks(
             step=32,
             value=1024,
         )
+        seed = gr.Slider(
+            label="Seed",
+            minimum=0,
+            maximum=MAX_SEED,
+            step=1,
+            value=0,
+        )
+
+    with gr.Row():
+        guidance_scale = gr.Slider(
+            label="Guidance scale",
+            minimum=1,
+            maximum=20,
+            step=0.1,
+            value=3.5,
+        )
+        num_inference_steps = gr.Slider(
+            label="Steps",
+            minimum=10,
+            maximum=100,
+            step=1,
+            value=50,
+        )
+
+    t2i_gen_bttn = gr.Button("Generate")
+
+    t2i_result = gr.Image(
+        label="Result",
+        show_label=False
+    )
+
+    gr.on(
+        triggers=[
+            t2i_gen_bttn.click
+        ],
+        fn=lambda: gr.update(interactive=False, value="Generating..."),
+        outputs=t2i_gen_bttn
+    ).then(
+        fn=t2i_gen,
+        inputs=[
+            prompt,
+            seed,
+            width,
+            height,
+            guidance_scale,
+            num_inference_steps
+        ],
+        outputs=t2i_result
+    ).then(
+        fn=lambda: gr.update(interactive=True, value="Generate"),
+        outputs=t2i_gen_bttn
 
 if __name__ == "__main__":
     demo.queue(max_size=20).launch(show_error=True)