zhiweili commited on
Commit
f614a18
Β·
1 Parent(s): e4c1258

print time cost

Browse files
Files changed (1) hide show
  1. app_onediff.py +7 -4
app_onediff.py CHANGED
@@ -49,7 +49,7 @@ base_pipe.unet = nexfort_compile(base_pipe.unet)
49
 
50
  def create_demo() -> gr.Blocks:
51
 
52
- @spaces.GPU(duration=10)
53
  def text_to_image(
54
  prompt:str,
55
  steps:int,
@@ -62,7 +62,7 @@ def create_demo() -> gr.Blocks:
62
  num_inference_steps=steps,
63
  ).images[0]
64
  run_task_time, time_cost_str = get_time_cost(run_task_time, time_cost_str)
65
- return generated_image
66
 
67
  def get_time_cost(run_task_time, time_cost_str):
68
  now_time = int(time.time()*1000)
@@ -84,12 +84,15 @@ def create_demo() -> gr.Blocks:
84
  g_btn = gr.Button("Generate")
85
 
86
  with gr.Row():
87
- generated_image = gr.Image(label="Generated Image", type="pil", interactive=False)
 
 
 
88
 
89
  g_btn.click(
90
  fn=text_to_image,
91
  inputs=[prompt, steps],
92
- outputs=[generated_image],
93
  )
94
 
95
  return demo
 
49
 
50
  def create_demo() -> gr.Blocks:
51
 
52
+ @spaces.GPU(duration=30)
53
  def text_to_image(
54
  prompt:str,
55
  steps:int,
 
62
  num_inference_steps=steps,
63
  ).images[0]
64
  run_task_time, time_cost_str = get_time_cost(run_task_time, time_cost_str)
65
+ return generated_image, time_cost_str
66
 
67
  def get_time_cost(run_task_time, time_cost_str):
68
  now_time = int(time.time()*1000)
 
84
  g_btn = gr.Button("Generate")
85
 
86
  with gr.Row():
87
+ with gr.Column:
88
+ generated_image = gr.Image(label="Generated Image", type="pil", interactive=False)
89
+ with gr.Column:
90
+ time_cost = gr.Textbox(label="Time Cost", lines=1, interactive=False)
91
 
92
  g_btn.click(
93
  fn=text_to_image,
94
  inputs=[prompt, steps],
95
+ outputs=[generated_image, time_cost],
96
  )
97
 
98
  return demo