Updated time output

app.py

@@ -29,7 +29,7 @@ def generate(image, option):
     start_time = time.time()
     output = pipe(prompt, num_inference_steps=50, output_type="pil")
     elapsed_time = time.time() - start_time
-    return (output.images[0], f"Inference time: {elapsed_time}")
+    return (output.images[0], "{:10.4f}".format(elapsed_time))
 
 examples = ["cartoon bird",
             "a drawing of a green pokemon with red eyes",
@@ -40,7 +40,7 @@ gr.Interface(
     inputs=[gr.inputs.Textbox(placeholder="cartoon bird", label="Prompt", lines=1),
             gr.inputs.Dropdown(choices=[option for option in pipes.keys()], default="Merged and quantized", label="Model version"),
            ],
-    outputs=[gr.outputs.Image(type="pil", label="Generated Image"), "text"],
+    outputs=[gr.outputs.Image(type="pil", label="Generated Image"), gr.outputs.Textbox(label="Inference time")],
     title="OpenVINO-optimized Stable Diffusion",
     description="This is the Optimum-based demo for NNCF-optimized Stable Diffusion pipeline trained on 'lambdalabs/pokemon-blip-captions' dataset and running with OpenVINO.\n"
                  "The pipeline is run using 8 vCPUs (4 cores) only.",