Fix single file caption error.

fireicewolf · Aug 29, 2024 · d9a12c9 · d9a12c9
1 parent e026f3f
commit d9a12c9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/joy.py b/utils/joy.py
@@ -92,7 +92,7 @@ def inference(self):
             [image for image in glob.glob(path_to_find, recursive=self.args.recursive)
              if image.lower().endswith(SUPPORT_IMAGE_FORMATS)]),
             key=lambda filename: (os.path.splitext(filename)[0])
-        ) if not os.path.isfile(self.args.data_path) else str(self.args.data_path) \
+        ) if not os.path.isfile(self.args.data_path) else [str(self.args.data_path)] \
             if str(self.args.data_path).lower().endswith(SUPPORT_IMAGE_FORMATS) else None
 
         if image_paths is None: