Update app.py

app.py

@@ -21,15 +21,15 @@ num_steps=50,
     prior_cf_scale=1,
     prior_steps='25',
 """
-def infer(prompt, negative='low quality, bad quality'):
+def infer(prompt, negative='low quality, bad quality', w=768, h=768, sampler="ddim_sampler", steps=50, scale=4):
     images = model.generate_text2img(prompt, 
                            negative_prior_prompt=negative,
                            negative_decoder_prompt=negative, 
-                           num_steps=50,
+                           num_steps=steps,
                            batch_size=1,
-                           guidance_scale=4,
-                           h=768, w=768,
-                           sampler='ddim_sampler',
+                           guidance_scale=scale,
+                           h=h, w=w,
+                           sampler=sampler,
                            prior_cf_scale=1,
                            prior_steps="25",)
     return images
@@ -218,7 +218,12 @@ Kandinsky 2.1 was trained on a large-scale image-text dataset LAION HighRes and
                     border=(True, False, True, True),
                     rounded=(True, False, False, True),
                     container=False,
-                )
+                    )
+                width = gr.Number(label="image width", show_label=True, value=768)
+                height = gr.Number(label="image height", show_label=True, value=768)
+                sampler = gr.Radio(["ddim_sampler", "plms_sampler"], label="Sampler")
+                steps = gr.Number(label="steps", show_label=True, value=50)
+                scale = gr.Number(label="scale", show_label=True, value=50)
                 btn = gr.Button("Run").style(
                     margin=False,
                     rounded=(False, True, True, False),
@@ -228,11 +233,11 @@ Kandinsky 2.1 was trained on a large-scale image-text dataset LAION HighRes and
             grid=[2], height="auto"
         )
 
-        ex = gr.Examples(examples=examples, fn=infer, inputs=[text, negative], outputs=gallery, cache_examples=True)
+        ex = gr.Examples(examples=examples, fn=infer, inputs=[text, negative, width, height, sampler, steps, scale], outputs=gallery, cache_examples=True)
         ex.dataset.headers = [""]
 
-        text.submit(infer, inputs=[text, negative], outputs=gallery)
-        btn.click(infer, inputs=[text, negative], outputs=gallery)
+        text.submit(infer, inputs=[text, negative, width, height, sampler, steps, scale], outputs=gallery)
+        btn.click(infer, inputs=[text, negative, width, height, sampler, steps, scale], outputs=gallery)
 gr.Markdown("""