Enables image drop-in (#12)

2025-06-26 18:27:43 +00:00 · 2024-03-13 09:16:32 +01:00
parent 8d4d9a6ccf
commit 86a309683b
1 changed files with 1 additions and 0 deletions
--- a/cli_chat.py
+++ b/cli_chat.py
@@ -116,6 +116,7 @@ def chat(args, tokenizer, vl_chat_processor, vl_gpt, generation_config):
                while cur_img_idx < num_images:
                    try:
                        image_file = input(f"({cur_img_idx + 1}/{num_images}) Input the image file path: ")
+                        image_file = image_file.strip() # trim whitespaces around path, enables drop-in from for example Dolphin

                    except KeyboardInterrupt:
                        print()