guilhermewerner/image-caption-api
1
0
Fork 0
mirror of https://github.com/guilhermewerner/image-caption-api synced 2025-06-15 14:35:13 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update main.py

Browse Source
This commit is contained in:
Guilherme Werner
2023-10-06 22:21:57 -03:00
parent f926b7f4f5
commit 9d492be606
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
main.py
View File

@ -15,9 +15,9 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
print(device) print(device)
# Carregar o modelo, tokenizer e processador de imagem # Carregar o modelo, tokenizer e processador de imagem
finetuned_model = VisionEncoderDecoderModel.from_pretrained("nlpconnect/vit-gpt2-image-captioning").to(device) finetuned_model = VisionEncoderDecoderModel.from_pretrained("Trabalho/vit-swin-base-224-gpt2-image-captioning").to(device)
finetuned_tokenizer = GPT2TokenizerFast.from_pretrained("nlpconnect/vit-gpt2-image-captioning") finetuned_tokenizer = GPT2TokenizerFast.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
finetuned_image_processor = ViTImageProcessor.from_pretrained("nlpconnect/vit-gpt2-image-captioning") finetuned_image_processor = ViTImageProcessor.from_pretrained("Trabalho/vit-swin-base-224-gpt2-image-captioning")
# Função para carregar uma imagem # Função para carregar uma imagem
def load_image(image_path): def load_image(image_path):