Update README.md

README.md

@@ -13,6 +13,8 @@ tags:
 - bio-inspired
 - text-generation-inference
 - materials science
+base_model:
+- meta-llama/Llama-3.2-11B-Vision-Instruct
 pipeline_tag: image-text-to-text
 inference:
   parameters:
@@ -21,6 +23,7 @@ widget:
 - messages:
   - role: user
     content: <|image_1|>Can you describe what you see in the image?
+    
 ---
 ## Model Summary
 
@@ -62,9 +65,16 @@ The raw input text is:
 
 ### Sample inference code
 
+Update your transformers installation if necessary:
+```
+pip install -U transformers
+```
+
 This code snippets show how to get quickly started on a GPU:
 
 ```python
+from transformers import MllamaForConditionalGeneration, AutoProcessor
+
 DEVICE='cuda:0'
 model_id='lamm-mit/Cephalo-Llama-3.2-11B-Vision-Instruct-128k'