Update app.py

app.py

@@ -68,7 +68,7 @@ class calculateDuration:
     def __enter__(self):
         self.start_time = time.time()
         return self
-    
+
     def __exit__(self, exc_type, exc_value, traceback):
         self.end_time = time.time()
         self.elapsed_time = self.end_time - self.start_time
@@ -137,13 +137,13 @@ def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, wid
     # Load LoRA weights
     with calculateDuration(f"Loading LoRA weights for {selected_lora['title']}"):
         if "weights" in selected_lora:
-            pipe.load_lora_weights(lora_path, weight_name=selected_lora["weights"], adapter_name="soon")
-            pipe.load_lora_weights("ostris/OpenFLUX.1", weight_name="openflux1-v0.1.0-fast-lora.safetensors", adapter_name="fast")
-            pipe.set_adapters(["fast", "soon"], adapter_weights=[1.0, lora_scale])
+            pipe.load_lora_weights(lora_path, weight_name=selected_lora["weights"], adapter_name=selected_lora['title'], adapter_weights=lora_scale)
+            pipe.load_lora_weights("ostris/OpenFLUX.1", weight_name="openflux1-v0.1.0-fast-lora.safetensors", adapter_name="fast", adapter_weights=[1.0])
+            pipe.set_adapters(["fast", 'title'], adapter_weights=[1.0, lora_scale])
         else:
-            pipe.load_lora_weights(lora_path, adapter_name="soon")
-            pipe.load_lora_weights("ostris/OpenFLUX.1", weight_name="openflux1-v0.1.0-fast-lora.safetensors", adapter_name="fast")
-            pipe.set_adapters(["fast", "soon"], adapter_weights=[1.0, lora_scale])
+            pipe.load_lora_weights(lora_path, adapter_name=selected_lora['title'], adapter_weights=lora_scale)
+            pipe.load_lora_weights("ostris/OpenFLUX.1", weight_name="openflux1-v0.1.0-fast-lora.safetensors", adapter_name="fast", adapter_weights=[1.0])
+            pipe.set_adapters(["fast", 'title'], adapter_weights=[1.0, lora_scale])
         
     # Set random seed for reproducibility
     with calculateDuration("Randomizing seed"):