Update app.py

app.py

@@ -30,13 +30,13 @@ pipe = CogVideoXImageToVideoPipeline.from_pretrained(
     torch_dtype=torch.bfloat16)
 
 def infer(prompt, image_path, orbit_type, progress=gr.Progress(track_tqdm=True)):
-    lora_path = None
+    lora_path = "checkpoints/"
     if orbit_type == "Left":
-        lora_path = "checkpoints/orbit_left_lora_weights.safetensors"
+        weight_name = "orbit_left_lora_weights.safetensors"
     elif orbit_type == "Up":
-        lora_path = "checkpoints/orbit_up_lora_weights.safetensors"
+        weight_name = "orbit_up_lora_weights.safetensors"
     lora_rank = 256
-    pipe.load_lora_weights(lora_path, weight_name="pytorch_lora_weights.safetensors", adapter_name="test_1")
+    pipe.load_lora_weights(lora_path, weight_name=weight_name, adapter_name="test_1")
     pipe.fuse_lora(lora_scale=1 / lora_rank)
     pipe.to("cuda")