Update app.py

app.py

@@ -194,7 +194,7 @@ def run_flux(
                 num_inference_steps=num_inference_steps_slider,
             ).images[0]
     progress(99, "Generate image success!")
-    pipe.unload_lora_weights()
+    
     return generated_image
 
 
@@ -325,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: