Update README.md

README.md

@@ -81,19 +81,18 @@ from PIL import Image
 from transformers import (
   LlavaForConditionalGeneration,
   AutoTokenizer,
+  AutoProcessor,
   CLIPImageProcessor
 )
 #In this repo, needed for version < 4.41.1
 #from processing_llavagemma import LlavaGemmaProcessor
+#processor = LlavaGemmaProcessor( tokenizer=AutoTokenizer.from_pretrained(checkpoint), image_processor=CLIPImageProcessor.from_pretrained(checkpoint))
 
 checkpoint = "Intel/llava-gemma-2b"
 
 # Load model
 model = LlavaForConditionalGeneration.from_pretrained(checkpoint)
-processor = LlavaGemmaProcessor(
-    tokenizer=AutoTokenizer.from_pretrained(checkpoint),
-    image_processor=CLIPImageProcessor.from_pretrained(checkpoint)
-)
+processor = AutoProcessor.from_pretrained(checkpoint)
 
 # Prepare inputs
 # Use gemma chat template