Update README.md

README.md

@@ -72,11 +72,11 @@ Users should:
 ## How to Get Started with the Model
 
 ```python
-from transformers import AutoProcessor, AutoModelForVision2Seq
+from transformers import AutoProcessor, LlavaForConditionalGeneration
 from PIL import Image
 
 processor = AutoProcessor.from_pretrained("ServiceNow/Pixtral-12B-2409-StarFlow")
-model = AutoModelForVision2Seq.from_pretrained("ServiceNow/Pixtral-12B-2409-StarFlow")
+model = LlavaForConditionalGeneration.from_pretrained("ServiceNow/Pixtral-12B-2409-StarFlow")
 
 image = Image.open("workflow_sketch.png")
 inputs = processor(images=image, text="Generate workflow JSON", return_tensors="pt")