Update README.md

README.md

@@ -17,7 +17,7 @@ class BLIPNet(torch.nn.Module):
     def __init__(self):
         super().__init__()
         # Generation Model
-        self.model = BlipForConditionalGeneration.from_pretrained("MODEL_NAME", cache_dir="model")
+        self.model = BlipForConditionalGeneration.from_pretrained("Salesforceblip-image-captioning-base", cache_dir="model")
         # Same with https://huggingface.co/uf-aice-lab/BLIP-Math
         self.ebd_dim = 443136