jdytnty

app.py

@@ -30,8 +30,8 @@ id_model = Idefics3ForConditionalGeneration.from_pretrained("HuggingFaceM4/Idefi
         #_attn_implementation="flash_attention_2"
                                                         ).to("cuda")
 
-BAD_WORDS_IDS = processor.tokenizer(["<image>", "<fake_token_around_image>"], add_special_tokens=False).input_ids
-EOS_WORDS_IDS = [processor.tokenizer.eos_token_id]
+BAD_WORDS_IDS = id_processor.tokenizer(["<image>", "<fake_token_around_image>"], add_special_tokens=False).input_ids
+EOS_WORDS_IDS = [id_processor.tokenizer.eos_token_id]
 
 @spaces.GPU
 def model_inference(