Update README.md

README.md

@@ -132,8 +132,14 @@ The curves show an increasing trend of stress with strain, indicating that the m
 Next we provide a convenience function for inference. This function takes the model, processor, question, and images, along with messages and images objects for repeated chat-like interactions with the model.
 
 ```python
+from tqdm.notebook import tqdm
 from transformers.image_utils import load_image
 
+def ensure_list(obj):
+    if not isinstance(obj, list):
+        return [obj]
+    return obj
+
 def is_url_or_filename(val) -> bool:
     # Check if it's a URL
     if isinstance(val, str):