blip-base / README.md
DevJGraham's picture
Create README.md - Add code for uploading model and processor
a5b6d87 verified
from transformers import BlipProcessor, BlipModel
# Load model & processor
processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
model = BlipModel.from_pretrained("Salesforce/blip-image-captioning-base")
from huggingface_hub import notebook_login
notebook_login()
processor.save_pretrained('blip-base') # safe_serialization=True
model.save_pretrained('blip-base')
!hf repo create blip-base --repo-type model
from huggingface_hub import upload_folder
upload_folder(
repo_id='DevJGraham/blip-base',
folder_path = 'blip-base'
)