catch error

serve/vote_utils.py

@@ -10,6 +10,7 @@ from .log_utils import build_logger
 from .constants import IMAGE_DIR, VIDEO_DIR
 import imageio
 from diffusers.utils import load_image
+import torch
 
 ig_logger = build_logger("gradio_web_server_image_generation", "gr_web_image_generation.log") # ig = image generation, loggers for single model direct chat
 igm_logger = build_logger("gradio_web_server_image_generation_multi", "gr_web_image_generation_multi.log") # igm = image generation multi, loggers for side-by-side and battle
@@ -144,10 +145,14 @@ def vote_last_response_vgm(states, vote_type, model_selectors, request: gr.Reque
             r = requests.get(state.output)
             with open(output_file, 'wb') as outfile:
                 outfile.write(r.content)
-        else:
+        elif isinstance(state.output, torch.Tensor):
             print("======== video shape: ========")
             print(state.output.shape)
             imageio.mimwrite(output_file, state.output, fps=8, quality=9)
+        else:
+            r = requests.get(state.output)
+            with open(output_file, 'wb') as outfile:
+                outfile.write(r.content)
         save_video_file_on_log_server(output_file)