Update README.md

README.md

@@ -21,8 +21,8 @@ library_name: transformers
 ## Model fine-tuning 🏋️‍
 
 - Trained for 10 epochs
-- lr:  5e−5,
-- Adam optimizer,
+- lr:  5e−5
+- Adam optimizer
 - half-precision (fp16)
 
 ## Test set metrics 🧾
@@ -38,8 +38,8 @@ import requests
 from PIL import Image
 from transformers import BlipProcessor, BlipForConditionalGeneration
 
-processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
-model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base").to("cuda")
+processor = BlipProcessor.from_pretrained("michelecafagna26/blip-base-captioning-ft-hl-scenes")
+model = BlipForConditionalGeneration.from_pretrained("michelecafagna26/blip-base-captioning-ft-hl-scenes").to("cuda")
 
 img_url = 'https://datasets-server.huggingface.co/assets/michelecafagna26/hl/--/default/train/0/image/image.jpg' 
 raw_image = Image.open(requests.get(img_url, stream=True).raw).convert('RGB')