Update app.py

app.py

@@ -32,6 +32,7 @@ JS = """function () {
 
 if torch.cuda.is_available():
     pipe = FluxPipeline.from_pretrained(model, torch_dtype=torch.bfloat16)
+    pipe.enable_model_cpu_offload()
 
 
 def scrape_lora_link(url):
@@ -59,13 +60,14 @@ def scrape_lora_link(url):
         return None
 
 def enable_lora(lora_scale, lora_in, lora_add):
+    pipe.unload_lora_weights()
     if not lora_in and not lora_add:
         return
     else:
         if lora_add:
             lora_in = lora_add
         url = f'https://huggingface.co/{lora_in}/tree/main'
-        lora_name = scrape_lora_link(url)    
+        lora_name = scrape_lora_link(url)
         pipe.load_lora_weights(lora_in, weight_name=lora_name)
         pipe.fuse_lora(lora_scale=lora_scale)