Update README.md

README.md

@@ -24,8 +24,7 @@ npm i @huggingface/transformers
 import {
     Florence2ForConditionalGeneration,
     AutoProcessor,
-    AutoTokenizer,
-    RawImage,
+    load_image,
 } from '@huggingface/transformers';
 
 // Load model, processor, and tokenizer
@@ -39,27 +38,26 @@ const model = await Florence2ForConditionalGeneration.from_pretrained(model_id,
     },
 });
 const processor = await AutoProcessor.from_pretrained(model_id);
-const tokenizer = await AutoTokenizer.from_pretrained(model_id);
 
 // Load image and prepare vision inputs
 const url = 'https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/transformers/tasks/car.jpg';
-const image = await RawImage.fromURL(url);
-const vision_inputs = await processor(image);
+const image = await load_image(url);
 
 // Specify task and prepare text inputs
 const task = '<MORE_DETAILED_CAPTION>';
 const prompts = processor.construct_prompts(task);
-const text_inputs = tokenizer(prompts);
+
+// Pre-process the image and text inputs
+const inputs = await processor(image, prompts);
 
 // Generate text
 const generated_ids = await model.generate({
-    ...text_inputs,
-    ...vision_inputs,
+    ...inputs,
     max_new_tokens: 256,
 });
 
 // Decode generated text
-const generated_text = tokenizer.batch_decode(generated_ids, { skip_special_tokens: false })[0];
+const generated_text = processor.batch_decode(generated_ids, { skip_special_tokens: false })[0];
 
 // Post-process the generated text
 const result = processor.post_process_generation(generated_text, task, image.size);