ZhangYuhan commited on
Commit
7bb496b
1 Parent(s): 32dec53

update serve

Browse files
Files changed (2) hide show
  1. serve/gradio_web_t2s.py +3 -2
  2. serve/inference.py +1 -1
serve/gradio_web_t2s.py CHANGED
@@ -63,9 +63,10 @@ Find out who is the 🥇conditional image generation models! More models are goi
63
  placeholder="👉 Enter your prompt or Sample a random prompt, and press ENTER",
64
  container=True,
65
  elem_id="input_box",
 
66
  interactive=False, ## only support offline
67
  )
68
- sample_btn = gr.Button(value="🎲 Sample", variant="primary", scale=0)
69
  with gr.Column():
70
  normal_left = gr.Image(width=512, label = "Normal A", show_download_button=True)
71
  rgb_left = gr.Image(width=512, label = "RGB A", show_download_button=True)
@@ -163,7 +164,7 @@ Find out who is the 🥇conditional image generation models! More models are goi
163
  clear_btn = gr.Button(value="🗑️ Clear", interactive=False)
164
  share_btn = gr.Button(value="📷 Share")
165
  regenerate_btn = gr.Button(value="🔄 Regenerate", interactive=False)
166
- send_btn = gr.Button(value="📤 Send", variant="primary", scale=0)
167
 
168
  gr.Markdown(acknowledgment_md, elem_id="ack_markdown")
169
 
 
63
  placeholder="👉 Enter your prompt or Sample a random prompt, and press ENTER",
64
  container=True,
65
  elem_id="input_box",
66
+ lines = 10,
67
  interactive=False, ## only support offline
68
  )
69
+ sample_btn = gr.Button(value="🎲 Sample", variant="primary")
70
  with gr.Column():
71
  normal_left = gr.Image(width=512, label = "Normal A", show_download_button=True)
72
  rgb_left = gr.Image(width=512, label = "RGB A", show_download_button=True)
 
164
  clear_btn = gr.Button(value="🗑️ Clear", interactive=False)
165
  share_btn = gr.Button(value="📷 Share")
166
  regenerate_btn = gr.Button(value="🔄 Regenerate", interactive=False)
167
+ send_btn = gr.Button(value="📤 Send", variant="primary")
168
 
169
  gr.Markdown(acknowledgment_md, elem_id="ack_markdown")
170
 
serve/inference.py CHANGED
@@ -430,7 +430,7 @@ def generate_t2s_multi_annoy(gen_func, render_func,
430
  }
431
  else:
432
  start_time = time.time()
433
- shape_0, shape_1, model_name_0, model_name_1 = gen_func(text, model_name_0, model_name_1, i2s_mode=False)
434
  generate_time = time.time() - start_time
435
 
436
  videos_0, videos_1 = render_func(shape_0, model_name_0, shape_1, model_name_1)
 
430
  }
431
  else:
432
  start_time = time.time()
433
+ shape_0, shape_1, model_name_0, model_name_1 = gen_func(text, model_name_0, model_name_1, i2s_model=False)
434
  generate_time = time.time() - start_time
435
 
436
  videos_0, videos_1 = render_func(shape_0, model_name_0, shape_1, model_name_1)