update

app.py

@@ -6,7 +6,7 @@ from models.vsa_prompt import COCO_CLASSES
 SAMPLES = {
     "images/iclr.jpg": ("What prize did this paper win in 2024?", ", ".join(COCO_CLASSES)),
     "images/tesla.jpg": ("What's the income of this company?", "car"),
-    "images/xiaomi.jpg": ("Provide information about the new products of this brand.", ", ".join(COCO_CLASSES)),
+    "images/dell.jpg": ("I want to replace this brand of laptop, please give me some suggestions on new products in 2024.", ", ".join(COCO_CLASSES)),
     "images/leshi.jpg": ("Provide information about new products of this brand of potato chips in 2024.", ", ".join(COCO_CLASSES)),
 }
 SAMPLE_IMAGES = list(SAMPLES.keys())

models/vsa_model.py

@@ -126,7 +126,7 @@ class VLM:
         load_8bit: bool = False,
         load_4bit: bool = True,
         temperature: float = 0.2,
-        max_new_tokens: int = 2000,
+        max_new_tokens: int = 1024,
     ):
         disable_torch_init()
         model_name = get_model_name_from_path(model_path)