update prompt ins

This commit is contained in:
ThuanNaN 2025-01-13 11:12:55 +07:00
parent 8415c675a5
commit 82016e32fc

View File

@ -35,7 +35,7 @@ class MVTecDataset(Dataset):
image = Image.open(image_path).convert("RGB")
image = self.vis_processor(image)
input = "detect a defect or not-defect object and return the bounding boxes and its label. If not, bound around the object."
input = "a defect or not-defect object and return the bounding boxes and its label. If not, bound around the object."
ans_defect = "defect" if info["is_broken"] == True else "non-defect"
ans_para = f"<p>{ans_cls}-{ans_defect}</p>"