Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import torch
 from diffusers import HunyuanVideoPipeline, HunyuanVideoTransformer3DModel
 from PIL import Image
@@ -5,7 +6,6 @@ import numpy as np
 import gradio as gr
 import os
 import gc
-import spaces
 
 
 model_id = "hunyuanvideo-community/HunyuanVideo"
@@ -40,6 +40,8 @@ def generate(prompt, width=832, height=832, num_inference_steps=30, lora_id=None
         return output
     finally:
         # Always clear memory, even if an error occurs
+        if lora_id and lora_id.strip() != "":
+            pipe.unload_lora_weights()
         torch.cuda.empty_cache()
         gc.collect()