Update README.md

README.md

@@ -222,8 +222,6 @@ print(answer)
 
 ## 🤗 Transformers:
 ```python
-from __future__ import annotations
-
 import torch
 from PIL import Image
 from transformers import AutoProcessor, Qwen2_5_VLForConditionalGeneration
@@ -244,7 +242,7 @@ model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
     trust_remote_code=True,
 )
 
-img = Image.open("invoice.png").convert("RGB")
+img = Image.open("image.png").convert("RGB")
 messages = [{
     "role": "user",
     "content": [
@@ -252,14 +250,14 @@ messages = [{
     ],
 }]
 prompt = processor.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
-enc = processor(text=prompt, images=[img], return_tensors="pt").to(model.device)
+model_input = processor(text=prompt, images=[img], return_tensors="pt").to(model.device)
 
 with torch.no_grad():
-    out = model.generate(**enc, temperature = 0.7, max_new_tokens=5000)
-
-out = processor.decode(out[0])
+    model_output = model.generate(**model_input, temperature = 0.7, max_new_tokens=5000)
 
-reasoning = out.split("<thinking>")[1].split("</thinking>")[0]
-answer  = out.split("<answer>")[1].split("</answer>")[0]
+result = processor.decode(model_output[0])
+reasoning = result.split("<think>")[1].split("</think>")[0]
+answer  = result.split("<answer>")[1].split("</answer>")[0]
+print(answer)
 ```