Update README.md

README.md

@@ -62,15 +62,24 @@ The model was fine-tuned on several datasets:
 
 - **Accuracy on Manual-VQA Tasks**: 30.34%
 
+## Required Libraries
+
+Before you start, ensure you have the following libraries installed:
+
+```
+pip install git+https://github.com/andimarafioti/transformers.git@idefics3
+```
+
 ## Usage
 To use the model with the Hugging Face `transformers` library:
 
 ```python
-from transformers import AutoTokenizer, AutoModel
+from transformers import AutoProcessor, Idefics3ForConditionalGeneration
+
+DEVICE = f"cuda" if torch.cuda.is_available() else 'cpu' if torch.cpu.is_available() else 'mps'
+display(DEVICE)
+if DEVICE == 'cuda': display(torch.cuda.device_count())
 
-# Load the tokenizer and model
-tokenizer = AutoTokenizer.from_pretrained("nectec/Pathumma-llm-vision-1.0.0")
-model = AutoModel.from_pretrained("nectec/Pathumma-llm-vision-1.0.0")
 N = 5
 
 processor = AutoProcessor.from_pretrained(