Update app.py

app.py

@@ -131,9 +131,25 @@ def get_video(prompt, seed, ddim_steps):
                           )
     return save_results(samples)
 
-def get_video_lora(prompt, seed, ddim_steps):
+def get_video_lora(prompt, seed, ddim_steps, model_choice):
+    
+    model_to_style = {
+        "Frozen": ", frozenmovie style",
+        "Coco": ", coco style",
+        "Loving Vincent": ", Loving Vincent style",
+        "MakotoShinkai YourName": ", MakotoShinkaiYourName style"
+    }
+    
+    model_to_index = {
+        "Frozen": 2
+        "Coco": 4,
+        "Loving Vincent": 1,
+        "MakotoShinkai YourName": 3
+    }
+    
     seed_everything(seed)
-    change_lora(model, inject_lora=True, lora_scale=1.0,lora_path = filename_list[2])
+    prompt = prompt + model_to_style[model_choice]
+    change_lora(model, inject_lora=True, lora_scale=1.0,lora_path = filename_list[model_to_index[model_choice]])
     samples = sample_text2video(model, prompt, n_samples = 1, batch_size = 1,
                           sampler=ddim_sampler, ddim_steps=ddim_steps
                           )