fix bug in gradio edition

easyanimate/ui/ui.py

@@ -6,6 +6,7 @@ import os
 import random
 import base64
 import requests
+import pkg_resources
 from datetime import datetime
 from glob import glob
 
@@ -37,6 +38,9 @@ scheduler_dict = {
     "DDIM": DDIMScheduler,
 }
 
+gradio_version = pkg_resources.get_distribution("gradio").version
+gradio_version_is_above_4 = True if int(gradio_version.split('.')[0]) >= 4 else False
+
 css = """
 .toolbutton {
     margin-buttom: 0em 0em 0em 0em;
@@ -726,12 +730,18 @@ class EasyAnimateController_EAS:
             save_sample_path = os.path.join(self.savedir_sample, prefix + f".png")
             with open(save_sample_path, "wb") as file:
                 file.write(decoded_data)
-            return gr.Image.update(value=save_sample_path, visible=True), gr.Video.update(value=None, visible=False), "Success"
+            if gradio_version_is_above_4:
+                return gr.Image(value=save_sample_path, visible=True), gr.Video(value=None, visible=False), "Success"
+            else:
+                return gr.Image.update(value=save_sample_path, visible=True), gr.Video.update(value=None, visible=False), "Success"
         else:
             save_sample_path = os.path.join(self.savedir_sample, prefix + f".mp4")
             with open(save_sample_path, "wb") as file:
                 file.write(decoded_data)
-            return gr.Image.update(visible=False, value=None), gr.Video.update(value=save_sample_path, visible=True), "Success"
+            if gradio_version_is_above_4:
+                return gr.Image(visible=False, value=None), gr.Video(value=save_sample_path, visible=True), "Success"
+            else:
+                return gr.Image.update(visible=False, value=None), gr.Video.update(value=save_sample_path, visible=True), "Success"
 
 
 def ui_eas(edition, config_path, model_name, savedir_sample):