Update app.py

app.py

@@ -99,11 +99,12 @@ def remove_id_and_ext(text):
     return text
 
 def place_lora(current_prompt, lorabtn):
-    if f"<lora:{lorabtn}:1.0>" in current_prompt:
-        result = current_prompt.replace(f"<lora:{lorabtn}:1.0>", "")
-    else:
-        result = f"{current_prompt} <lora:{lorabtn}:1>"
-    return result
+  pattern = r"<lora:" + lora_name + r":.*?>"
+  
+  if re.search(pattern, current_prompt):
+    return re.sub(pattern, "", current_prompt)
+  else:
+    return current_prompt + "<lora:" + lora_name + ":1>"
 
 def create_grid(image_urls):
     # Download first image to get size