Update multi_script.py

multi_script.py

@@ -127,7 +127,17 @@ conv.append_message(conv.roles[0], qs)
 conv.append_message(conv.roles[1], None)
 prompt = conv.get_prompt()
 
-image_files = image_parser(args)
+if ".jpg" in image_file or ".png" in image_file:
+    image_files = image_parser(args)
+else:
+    import glob
+    import os
+    image_ext = ("*.png", '*.jpg')
+    image_files = []
+
+    for ext in image_ext:
+        image_files.extend(glob.glob(os.path.join(image_file, ext)))
+        
 images = load_images(image_files)
 image_sizes = [x.size for x in images]
 images_tensor = process_images(