Added seed

app.py

@@ -28,10 +28,11 @@ pipes = {
     "OpenVINO merged and quantized": pipe_tome_int8
 }
 
-def generate(prompt, option):
+def generate(prompt, option, seed):
     pipe = pipes[option]
+    set_seed(int(seed))
     start_time = time.time()
-    output = pipe(prompt, num_inference_steps=50, output_type="pil")
+    output = pipe(prompt, num_inference_steps=50, output_type="pil", height=512, width=512)
     elapsed_time = time.time() - start_time
     return (output.images[0], "{:10.4f}".format(elapsed_time))
 
@@ -45,6 +46,7 @@ gr.Interface(
     fn=generate,
     inputs=[gr.inputs.Textbox(placeholder="cartoon bird", label="Prompt", lines=1),
             gr.inputs.Dropdown(choices=model_options, default=model_options[-1], label="Model version"),
+            gr.inputs.Textbox(placeholder="42", label="Seed", lines=1)
            ],
     outputs=[gr.outputs.Image(type="pil", label="Generated Image"), gr.outputs.Textbox(label="Inference time")],
     title="OpenVINO-optimized Stable Diffusion",