Update README.md

README.md

@@ -14,18 +14,21 @@ pipeline_tag: image-to-text
 
 Usage is as follows:
 
-```
+```python
 from transformers import InstructBlipProcessor, InstructBlipForConditionalGeneration
 import torch
 from PIL import Image
 import requests
+
 model = InstructBlipForConditionalGeneration.from_pretrained("UBC-NLP/Peacock")
 processor = InstructBlipProcessor.from_pretrained("UBC-NLP/Peacock")
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model.to(device)
+
 url = "https://upload.wikimedia.org/wikipedia/commons/8/83/Socotra_dragon_tree.JPG"
 image = Image.open(requests.get(url, stream=True).raw).convert("RGB")
 prompt = "اوصف الصوره"
+
 inputs = processor(images=image, text=prompt, return_tensors="pt").to(device)
 outputs = model.generate(
         **inputs,