Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
8a702cc
1
Parent(s):
18d0376
fix bug
Browse files- serve/vote_utils.py +3 -3
serve/vote_utils.py
CHANGED
|
@@ -609,7 +609,7 @@ def generate_igm_annoy(gen_func, state0, state1, text, model_name0, model_name1,
|
|
| 609 |
state1.model_name = model_name1
|
| 610 |
|
| 611 |
yield state0, state1, generated_image0, generated_image1, \
|
| 612 |
-
gr.Markdown(f"### Model A: {model_name0}"), gr.Markdown(f"### Model B: {model_name1}")
|
| 613 |
|
| 614 |
finish_tstamp = time.time()
|
| 615 |
# logger.info(f"===output===: {output}")
|
|
@@ -813,7 +813,7 @@ def generate_iem_annoy(gen_func, state0, state1, source_text, target_text, instr
|
|
| 813 |
state1.model_name = model_name1
|
| 814 |
|
| 815 |
yield state0, state1, generated_image0, generated_image1, \
|
| 816 |
-
gr.Markdown(f"### Model A: {model_name0}"), gr.Markdown(f"### Model B: {model_name1}")
|
| 817 |
|
| 818 |
finish_tstamp = time.time()
|
| 819 |
# logger.info(f"===output===: {output}")
|
|
@@ -1046,4 +1046,4 @@ def generate_vgm_annoy(gen_func, state0, state1, text, model_name0, model_name1,
|
|
| 1046 |
save_video_file_on_log_server(output_file)
|
| 1047 |
|
| 1048 |
yield state0, state1, f'{VIDEO_DIR}/generation/{state0.conv_id}.mp4', f'{VIDEO_DIR}/generation/{state1.conv_id}.mp4', \
|
| 1049 |
-
gr.Markdown(f"### Model A: {model_name0}"), gr.Markdown(f"### Model B: {model_name1}")
|
|
|
|
| 609 |
state1.model_name = model_name1
|
| 610 |
|
| 611 |
yield state0, state1, generated_image0, generated_image1, \
|
| 612 |
+
gr.Markdown(f"### Model A: {model_name0}", visible=False), gr.Markdown(f"### Model B: {model_name1}", visible=False)
|
| 613 |
|
| 614 |
finish_tstamp = time.time()
|
| 615 |
# logger.info(f"===output===: {output}")
|
|
|
|
| 813 |
state1.model_name = model_name1
|
| 814 |
|
| 815 |
yield state0, state1, generated_image0, generated_image1, \
|
| 816 |
+
gr.Markdown(f"### Model A: {model_name0}", visible=False), gr.Markdown(f"### Model B: {model_name1}", visible=False)
|
| 817 |
|
| 818 |
finish_tstamp = time.time()
|
| 819 |
# logger.info(f"===output===: {output}")
|
|
|
|
| 1046 |
save_video_file_on_log_server(output_file)
|
| 1047 |
|
| 1048 |
yield state0, state1, f'{VIDEO_DIR}/generation/{state0.conv_id}.mp4', f'{VIDEO_DIR}/generation/{state1.conv_id}.mp4', \
|
| 1049 |
+
gr.Markdown(f"### Model A: {model_name0}", visible=False), gr.Markdown(f"### Model B: {model_name1}", visible=False)
|