Spaces:
Running
Running
ZhangYuhan
commited on
Commit
•
7bb496b
1
Parent(s):
32dec53
update serve
Browse files- serve/gradio_web_t2s.py +3 -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"
|
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"
|
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,
|
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)
|