diff --git a/cli_demo.py b/cli_demo.py index 62539e7..a931595 100644 --- a/cli_demo.py +++ b/cli_demo.py @@ -241,7 +241,11 @@ def main(): try: if first: if '' not in query: - query = f'{"" * len(image_paths)} ' + query + # query = f'{"" * len(image_paths)} ' + query + img_query = "" + for _ in image_paths: + img_query += '' + query = img_query + query first = False if args.cpu_only: model.bfloat16()