print time cost

app_onediff.py

@@ -49,7 +49,7 @@ base_pipe.unet = nexfort_compile(base_pipe.unet)
 
 def create_demo() -> gr.Blocks:
 
-    @spaces.GPU(duration=10)
+    @spaces.GPU(duration=30)
     def text_to_image(
         prompt:str,
         steps:int,
@@ -62,7 +62,7 @@ def create_demo() -> gr.Blocks:
             num_inference_steps=steps,
         ).images[0]
         run_task_time, time_cost_str = get_time_cost(run_task_time, time_cost_str)
-        return generated_image
+        return generated_image, time_cost_str
 
     def get_time_cost(run_task_time, time_cost_str):
         now_time = int(time.time()*1000)
@@ -84,12 +84,15 @@ def create_demo() -> gr.Blocks:
                 g_btn = gr.Button("Generate")
                 
         with gr.Row():
-            generated_image = gr.Image(label="Generated Image", type="pil", interactive=False)
+            with gr.Column:
+                generated_image = gr.Image(label="Generated Image", type="pil", interactive=False)
+            with gr.Column:
+                time_cost = gr.Textbox(label="Time Cost", lines=1, interactive=False)
         
         g_btn.click(
             fn=text_to_image,
             inputs=[prompt, steps],
-            outputs=[generated_image],
+            outputs=[generated_image, time_cost],
         )
 
     return demo