Spaces:
Running
on
Zero
Running
on
Zero
tianleliphoebe
commited on
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)
|