Update app.py

app.py

@@ -172,7 +172,10 @@ def infer_with_lora(input_image, prompt, selected_index, custom_lora, seed=42, r
     input_image = input_image.convert("RGB")
     # Add trigger word to prompt
     trigger_word = lora_to_use["trigger_word"]
-    prompt = f"convert the style of this portrait photo to {trigger_word} while maintaining the identity of the person. {prompt}. make sure to maintain the person's facial identity and features, while still changing the style to {trigger_word}."
+    if trigger_word == "CCTUR3"
+        prompt = f"create a CCTUR3 caricature of the person in the photo {prompt}, maintain the facial identity of the person and general features while still turning it into a caricature"
+    else:
+        prompt = f"convert the style of this portrait photo to {trigger_word} while maintaining the identity of the person. {prompt}. Make sure to maintain the person's facial identity and features, while still changing the overall style to {trigger_word}."
     
     try:
         image = pipe(