From 9d492be606c2ac5a1b30db818627c47bad0a4b19 Mon Sep 17 00:00:00 2001 From: Guilherme Werner Date: Fri, 6 Oct 2023 22:21:57 -0300 Subject: [PATCH] Update main.py --- main.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/main.py b/main.py index efbdf16..8aa48a0 100644 --- a/main.py +++ b/main.py @@ -15,9 +15,9 @@ device = "cuda" if torch.cuda.is_available() else "cpu" print(device) # 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_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 def load_image(image_path):