fix a bug

serve/vote_utils.py

@@ -926,7 +926,7 @@ def generate_vgm(gen_func, state0, state1, text, model_name0, model_name1, reque
     state0.model_name = model_name0
     state1.model_name = model_name1
 
-    yield state0, state1, generated_video0, generated_video1
+    # yield state0, state1, generated_video0, generated_video1
     print("====== model name =========")
     print(state0.model_name)
     print(state1.model_name)
@@ -976,7 +976,7 @@ def generate_vgm(gen_func, state0, state1, text, model_name0, model_name1, reque
             print(state.output.shape)
             imageio.mimwrite(output_file, state.output, fps=8, quality=9)
         save_video_file_on_log_server(output_file)
-    # yield state0, state1, f'{VIDEO_DIR}/generation/{state0.conv_id}.mp4', f'{VIDEO_DIR}/generation/{state1.conv_id}.mp4'
+    yield state0, state1, f'{VIDEO_DIR}/generation/{state0.conv_id}.mp4', f'{VIDEO_DIR}/generation/{state1.conv_id}.mp4'
 
 
 def generate_vgm_annoy(gen_func, state0, state1, text, model_name0, model_name1, request: gr.Request):
@@ -999,8 +999,8 @@ def generate_vgm_annoy(gen_func, state0, state1, text, model_name0, model_name1,
     state0.model_name = model_name0
     state1.model_name = model_name1
     
-    yield state0, state1, generated_video0, generated_video1, \
-        gr.Markdown(f"### Model A: {model_name0}"), gr.Markdown(f"### Model B: {model_name1}")
+    # yield state0, state1, generated_video0, generated_video1, \
+    #     gr.Markdown(f"### Model A: {model_name0}"), gr.Markdown(f"### Model B: {model_name1}")
 
     finish_tstamp = time.time()
     # logger.info(f"===output===: {output}")
@@ -1044,5 +1044,5 @@ def generate_vgm_annoy(gen_func, state0, state1, text, model_name0, model_name1,
             imageio.mimwrite(output_file, state.output, fps=8, quality=9)
         save_video_file_on_log_server(output_file)
 
-    # yield state0, state1, f'{VIDEO_DIR}/generation/{state0.conv_id}.mp4', f'{VIDEO_DIR}/generation/{state1.conv_id}.mp4', \
-    #     gr.Markdown(f"### Model A: {model_name0}"), gr.Markdown(f"### Model B: {model_name1}")
+    yield state0, state1, f'{VIDEO_DIR}/generation/{state0.conv_id}.mp4', f'{VIDEO_DIR}/generation/{state1.conv_id}.mp4', \
+        gr.Markdown(f"### Model A: {model_name0}"), gr.Markdown(f"### Model B: {model_name1}")