Update app.py

app.py

@@ -147,7 +147,9 @@ def generate_image(
     image_token_id = pipe.tokenizer.convert_tokens_to_ids(pipe.trigger_word)
     input_ids = pipe.tokenizer.encode(prompt)
     if image_token_id not in input_ids:
-        raise gr.Error(f"Cannot find the trigger word '{pipe.trigger_word}' in text prompt! Please refer to step 2️⃣")
+        prompt = "img" + prompt
+        input_ids = pipe.tokenizer.encode(prompt)
+        # raise gr.Error(f"Cannot find the trigger word '{pipe.trigger_word}' in text prompt! Please refer to step 2️⃣")
 
     if input_ids.count(image_token_id) > 1:
         raise gr.Error(f"Cannot use multiple trigger words '{pipe.trigger_word}' in text prompt!")