Update custom_tools/image_description_tool.py

custom_tools/image_description_tool.py

@@ -71,7 +71,7 @@ class ImageDescriptionTool(BaseTool):
         self.model.to(self.device)
         self.max_length = 16
         self.num_beams = 4
-        self.gen_kwargs = {"max_length": self.max_length, "num_beams": self.num_beams}
+        self.gen_kwargs = {"max_length": self.max_length}  # no num_beams = greedy decoding
     
     def apply(self, image_path: str) -> str:
         try: