Update README.md

README.md

@@ -61,6 +61,9 @@ See also [notebooks](./notebooks).
 
 ```python
 import requests
+from PIL import Image
+
+import torch
 from transformers import AutoProcessor
 from git_llm.git_llama import GitLlamaForCausalLM
 
@@ -82,22 +85,22 @@ text = f"##Instruction: Please answer the following question concletely. ##Quest
 
 # do preprocessing
 inputs = processor(
-text,
-image,
-return_tensors="pt",
-truncation=True,
+    text,
+    image,
+    return_tensors="pt",
+    truncation=True,
 )
 inputs = {k: v.to(f"cuda:{device_id}") for k, v in inputs.items()}
 
 # set eos token
 eos_token_id_list = [
-processor.tokenizer.pad_token_id,
-processor.tokenizer.eos_token_id,
+    processor.tokenizer.pad_token_id,
+    processor.tokenizer.eos_token_id,
 ]
 
 # do inference
 with torch.no_grad():
-out = model.generate(**inputs, max_length=256, do_sample=False, temperature=0., eos_token_id=eos_token_id_list)
+    out = model.generate(**inputs, max_length=256, do_sample=False, temperature=0., eos_token_id=eos_token_id_list)
 
 # print result
 print(processor.tokenizer.batch_decode(out))