Update app.py

app.py

@@ -53,21 +53,17 @@ def clear_gpu():
 
 def infer(image_path, prompt, orbit_type, progress=gr.Progress(track_tqdm=True)):
     
-    
     lora_path = "checkpoints/"
     if orbit_type == "Left":
         weight_name = "orbit_left_lora_weights.safetensors"
+        adapter_name = "orbit_left_lora_weights"
     elif orbit_type == "Up":
         weight_name = "orbit_up_lora_weights.safetensors"
+        adapter_name = "orbit_lup_lora_weights"
     lora_rank = 256
 
-    pipe.unload_lora_weights()
-
-    # Generate a timestamp for adapter_name
-    adapter_timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-
     # Load LoRA weights on CPU, move to GPU afterward
-    pipe.load_lora_weights(lora_path, weight_name=weight_name, adapter_name=f"{adapter_timestamp}")
+    pipe.load_lora_weights(lora_path, weight_name=weight_name, adapter_name=adapter_name)
     pipe.fuse_lora(lora_scale=1 / lora_rank)
     
     # Move the pipeline to GPU for inference
@@ -87,6 +83,10 @@ def infer(image_path, prompt, orbit_type, progress=gr.Progress(track_tqdm=True))
         use_dynamic_cfg=True,
         generator=torch.Generator(device="cpu").manual_seed(seed)
     )
+
+    pipe.unfuse_lora()
+    pipe.unload_lora_weights()
+
     
     # Generate and save output video
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")