Update llava_llama3/serve/cli.py

llava_llama3/serve/cli.py

@@ -33,9 +33,9 @@ def chat_llava(args, image_file, text, tokenizer, model, image_processor, contex
     conv = conv_templates[args.conv_mode].copy()
     roles = conv.roles
     
-    print(image_file,type(image_file))
+    print(f"\033[91m{image_file}, {type(image_file)}\033[0m")
     image = load_image(image_file)
-    print(image,type(image))
+    print(f"\033[91m{image}, {type(image)}\033[0m")
     
     image_size = image.size
     image_tensor = process_images([image], image_processor, model.config)