Update README.md

README.md

@@ -51,10 +51,10 @@ processor = AutoProcessor.from_pretrained("visheratin/MC-LLaVA-3b", trust_remote
 
 with torch.inference_mode():
     inputs = processor(prompt, [raw_image], model, max_crops=100, num_tokens=728)
+    output = model.generate(**inputs, max_new_tokens=200, use_cache=True, do_sample=False,
+        eos_token_id=processor.tokenizer.eos_token_id, pad_token_id=processor.tokenizer.eos_token_id)
 
-output = model.generate(**inputs, max_new_tokens=200, use_cache=True, do_sample=False, eos_token_id=processor.tokenizer.eos_token_id, pad_token_id=processor.tokenizer.eos_token_id)
 result = processor.tokenizer.decode(output[0]).replace(prompt, "").replace("<|im_end|>", "")
-
 print(result)
 ```