From 86f70a3b0820e8f28fe6bdfdf674f1ba44b6ac89 Mon Sep 17 00:00:00 2001
From: Starrick Liu <73152103+StarrickLiu@users.noreply.github.com>
Date: Thu, 11 Sep 2025 18:54:25 +0800
Subject: [PATCH] [lint] format code with black (#22)

---
 scripts/vqa_inference.py | 17 ++++++++++++-----
 1 file changed, 12 insertions(+), 5 deletions(-)

diff --git a/scripts/vqa_inference.py b/scripts/vqa_inference.py
index 2e0d939..c3d2211 100644
--- a/scripts/vqa_inference.py
+++ b/scripts/vqa_inference.py
@@ -30,7 +30,12 @@ class VQAWrapper(object):
         return model
 
     def generate(self, image: Image.Image, text: str, **kwargs) -> str:
-        messages = [{"role": "user", "content": [{"type": "image"}, {"type": "text", "text": text}]}]
+        messages = [
+            {
+                "role": "user",
+                "content": [{"type": "image"}, {"type": "text", "text": text}],
+            }
+        ]
         text_prompt = self.processor.apply_chat_template(
             messages, tokenize=False, add_generation_prompt=True
         )
@@ -38,18 +43,19 @@ class VQAWrapper(object):
         inputs = {k: v.to(self.device) for k, v in inputs.items()}
 
         generation_params = {
-            "max_new_tokens": 1024, # default value, can be overridden by kwargs
+            "max_new_tokens": 1024,  # default value, can be overridden by kwargs
             "do_sample": False,
             "eos_token_id": self.processor.tokenizer.eos_token_id,
             "pad_token_id": self.processor.tokenizer.pad_token_id,
-            **kwargs
+            **kwargs,
         }
 
         with torch.no_grad():
             generated_ids = self.model.generate(**inputs, **generation_params)
 
         generated_ids = [
-            output_ids[len(input_ids):] for input_ids, output_ids in zip(inputs['input_ids'], generated_ids)
+            output_ids[len(input_ids) :]
+            for input_ids, output_ids in zip(inputs["input_ids"], generated_ids)
         ]
         response = self.processor.batch_decode(
             generated_ids, skip_special_tokens=True, clean_up_tokenization_spaces=False
@@ -67,9 +73,10 @@ if __name__ == "__main__":
 
         # img = Image.open("/path/to/your/local/image.jpg").convert("RGB")
         import requests
+
         img = Image.open(requests.get(test_image_url, stream=True).raw).convert("RGB")
         answer = wrapper.generate(img, test_question)
 
         print("model answer:", answer)
     except Exception as e:
-        print(f"model answer fail: {e}")
\ No newline at end of file
+        print(f"model answer fail: {e}")