Update app.py

app.py

@@ -15,6 +15,8 @@ from lvdm.utils.common_utils import str2bool
 from lvdm.utils.dist_utils import setup_dist, gather_data
 from lvdm.utils.saving_utils import npz_to_video_grid, npz_to_imgsheet_5d
 from utils import load_model, get_conditions, make_model_input_shape, torch_to_np
+from lvdm.models.modules.lora import change_lora
+
 from huggingface_hub import hf_hub_download
 
 config_path = "model_config.yaml"
@@ -129,7 +131,17 @@ def get_video(prompt, seed, ddim_steps):
                           )
     return save_results(samples)
 
+def get_video_lora(prompt, seed, ddim_steps):
+    seed_everything(seed)
+    change_lora(model, inject_lora=True, lora_scale=1.0,lora_path = filename_list[2])
+    samples = sample_text2video(model, prompt, n_samples = 1, batch_size = 1,
+                          sampler=ddim_sampler, ddim_steps=ddim_steps
+                          )
+    return save_results(samples)
+
+
 from gradio_t2v import create_demo as create_demo_basic
+from gradio_videolora import create_demo as create_demo_videolora
 
 DESCRIPTION = '# [Latent Video Diffusion Models](https://github.com/VideoCrafter/VideoCrafter)'
 DESCRIPTION += '\n<p>🤗🤗🤗 VideoCrafter is an open-source video generation and editing toolbox for crafting video content. This model can only be used for non-commercial purposes. To learn more about the model, take a look at the <a href="https://github.com/VideoCrafter/VideoCrafter" style="text-decoration: underline;" target="_blank">model card</a>.</p>'
@@ -139,6 +151,8 @@ with gr.Blocks(css='style.css') as demo:
     with gr.Tabs():
         with gr.TabItem('Basic Text2Video'):
             create_demo_basic(get_video)
+        with gr.TabItem('VideoLoRA'):
+            create_demo_videolora(get_video_lora)
 
 demo.queue(api_open=False).launch()