Steps parametrized.

app.py

@@ -59,10 +59,10 @@ def palette_to_attr(_palette):
     return _attr
 
 
-def generate(prompt, seed):
+def generate(prompt, seed, steps):
     generator = torch.Generator("cpu").manual_seed(int(seed))
     raw_image = \
-    pipe(prompt, height=image_height, width=image_width, num_inference_steps=20, generator=generator).images[0]
+    pipe(prompt, height=image_height, width=image_width, num_inference_steps=int(steps), generator=generator).images[0]
     palette = np.array(
         [[0, 0, 0], [0, 0, 255], [0, 255, 0], [0, 255, 255], [255, 0, 0], [255, 0, 255], [255, 255, 0],
          [255, 255, 255]])
@@ -104,7 +104,7 @@ def generate(prompt, seed):
 
 iface = gr.Interface(fn=generate,
                      title="ZX-Spectrum inspired images generator ",
-                     inputs=["text", "number"],
+                     inputs=["text", "number", "number"],
                      outputs=["image", "file"],
-                     examples=[["Cute cat", 123], ["Solar system", 123], ["Disco ball", 123]])
+                     examples=[["Cute cat", 123, 20], ["Solar system", 123, 20], ["Disco ball", 123, 20]])
 iface.launch()