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, MllamaForConditionalGeneration
 from PIL import Image
 
 processor = AutoProcessor.from_pretrained("ServiceNow/Llama-3.2-11B-Vision-Instruct-StarFlow")
-model = AutoModelForVision2Seq.from_pretrained("ServiceNow/Llama-3.2-11B-Vision-Instruct-StarFlow")
+model = MllamaForConditionalGeneration.from_pretrained("ServiceNow/Llama-3.2-11B-Vision-Instruct-StarFlow")
 
 image = Image.open("workflow_sketch.png")
 inputs = processor(images=image, text="Generate workflow JSON", return_tensors="pt")
@@ -222,4 +222,4 @@ Béchard, P., Wang, C., Abaskohi, A., Rodriguez, J., Pal, C., Vazquez, D., Gella
 ## Model Card Contact
 
 * Patrice Bechard - [[email protected]](mailto:[email protected])
-* ServiceNow Research – [research.servicenow.com](https://research.servicenow.com)
+* ServiceNow Research – [research.servicenow.com](https://research.servicenow.com)