tianleliphoebe commited on
Commit
8a702cc
1 Parent(s): 18d0376
Files changed (1) hide show
  1. 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)