Update app.py

app.py

@@ -194,6 +194,7 @@ def run_flux(
                 num_inference_steps=num_inference_steps_slider,
             ).images[0]
     progress(99, "Generate image success!")
+    
     return generated_image
 
 
@@ -324,7 +325,7 @@ def process(
         result["message"] = "generate image failed"
         print(e)
         generated_image = None
-
+    pipe.unload_lora_weights()
     clear_cuda_cache()
     print("run flux finish")
     if generated_image: