Update app.py

app.py

@@ -164,9 +164,9 @@ captioner = pipeline(model="ydshieh/vit-gpt2-coco-en",device='cuda', task="image
 captioner_2 = pipeline(model="Salesforce/blip-image-captioning-base",device='cuda', task="image-to-text")
 captioner_3 = pipeline(model="Salesforce/blip-image-captioning-large",device='cuda', task="image-to-text")
 #model5 = InstructBlipForConditionalGeneration.from_pretrained("Salesforce/instructblip-vicuna-7b").to(torch.bfloat16).to('cuda')
- #model5 = Blip2ForConditionalGeneration.from_pretrained("Salesforce/blip2-opt-2.7b-coco").to('cuda')
+model5 = Blip2ForConditionalGeneration.from_pretrained("Salesforce/blip2-opt-2.7b-coco").to('cuda')
 #processor5 = InstructBlipProcessor.from_pretrained("Salesforce/instructblip-vicuna-7b").to(torch.bfloat16).to('cuda')
- #processor5 = Blip2Processor.from_pretrained("Salesforce/blip2-opt-2.7b-coco")
+processor5 = Blip2Processor.from_pretrained("Salesforce/blip2-opt-2.7b-coco")
 txt_tokenizer = AutoTokenizer.from_pretrained(checkpoint, add_prefix_space=False)
 txt_tokenizer.tokenizer_legacy=False
 model = AutoModelForCausalLM.from_pretrained(checkpoint).to('cuda')