add requirements

app.py

@@ -1,11 +1,11 @@
 import time 
 
 from easyanimate.api.api import infer_forward_api, update_diffusion_transformer_api, update_edition_api
-from easyanimate.ui.ui import ui_modelscope, ui, ui_huggingface
+from easyanimate.ui.ui import ui_modelscope, ui_eas, ui
 
 if __name__ == "__main__":
     # Choose the ui mode  
-    ui_mode = "huggingface"
+    ui_mode = "eas"
     # Server ip
     server_name = "0.0.0.0"
     server_port = 7860
@@ -18,8 +18,8 @@ if __name__ == "__main__":
 
     if ui_mode == "modelscope":
         demo, controller = ui_modelscope(edition, config_path, model_name, savedir_sample)
-    elif ui_mode == "huggingface":
-        demo, controller = ui_huggingface(edition, config_path, model_name, savedir_sample)
+    elif ui_mode == "eas":
+        demo, controller = ui_eas(edition, config_path, model_name, savedir_sample)
     else:
         demo, controller = ui()
 

easyanimate/ui/ui.py

@@ -691,7 +691,7 @@ def post_eas(
     return outputs
 
 
-class EasyAnimateController_HuggingFace:
+class EasyAnimateController_EAS:
     def __init__(self, edition, config_path, model_name, savedir_sample):
         self.savedir_sample = savedir_sample
         os.makedirs(self.savedir_sample, exist_ok=True)
@@ -734,8 +734,8 @@ class EasyAnimateController_HuggingFace:
             return gr.Image.update(visible=False, value=None), gr.Video.update(value=save_sample_path, visible=True), "Success"
 
 
-def ui_huggingface(edition, config_path, model_name, savedir_sample):
-    controller = EasyAnimateController_HuggingFace(edition, config_path, model_name, savedir_sample)
+def ui_eas(edition, config_path, model_name, savedir_sample):
+    controller = EasyAnimateController_EAS(edition, config_path, model_name, savedir_sample)
 
     with gr.Blocks(css=css) as demo:
         gr.Markdown(

requirements.txt

@@ -0,0 +1,23 @@
+Pillow
+einops
+safetensors
+timm
+tomesd
+accelerate
+torch>=2.2.0
+torchdiffeq
+torchsde
+xformers
+decord
+datasets
+numpy
+scikit-image
+opencv-python
+omegaconf
+SentencePiece
+albumentations
+imageio[ffmpeg]
+imageio[pyav]
+tensorboard
+diffusers==0.27.0
+transformers==4.37.2