Update README.md

README.md

@@ -26,17 +26,17 @@ Fine-tuned Florence-2 model on DocumentVQA dataset to perform question answering
 # Get Started with the Model
 
 #### 1. Installation
-   
 ```python
-pip install torch transformers datasets flash_attn
+!pip install torch transformers datasets flash_attn
 ```
+
 #### 2. Loading model and processor
 ```python
 import torch
 from transformers import AutoModelForCausalLM, AutoProcessor
 
-model = AutoModelForCausalLM.from_pretrained("sahilnishad/Florence-2-FT-DocVQA")
-processor = AutoProcessor.from_pretrained("sahilnishad/Florence-2-FT-DocVQA")
+model = AutoModelForCausalLM.from_pretrained("sahilnishad/Florence-2-FT-DocVQA", trust_remote_code=True)
+processor = AutoProcessor.from_pretrained("sahilnishad/Florence-2-FT-DocVQA", trust_remote_code=True)
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model.to(device)
 ```