add trigger word

app.py

@@ -19,6 +19,8 @@ MAX_SEED = np.iinfo(np.int32).max
 IMAGE_SIZE = 1024
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 HF_TOKEN = os.environ.get("HF_TOKEN")
+trigger_word = "a_photo_of_TOK"
+
 login(token=HF_TOKEN)
 
 def remove_background(image: Image.Image, threshold: int = 50) -> Image.Image:
@@ -127,7 +129,7 @@ def process(
         seed_slicer = random.randint(0, MAX_SEED)
     generator = torch.Generator().manual_seed(seed_slicer)
     result = pipe(
-        prompt=input_text,
+        prompt=f"{input_text} {trigger_word}",
         image=resized_image,
         mask_image=resized_mask,
         width=width,