Update app.py

app.py

@@ -161,9 +161,9 @@ pipe = load_and_prepare_model()
 checkpoint = "microsoft/Phi-3.5-mini-instruct"
 captioner = pipeline(model="ydshieh/vit-gpt2-coco-en",device='cuda', task="image-to-text")
 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 = Blip2ForConditionalGeneration.from_pretrained("Salesforce/blip2-opt-2.7b-coco").to('cuda')
-processor5 = Blip2Processor.from_pretrained("Salesforce/blip2-opt-2.7b-coco")
+captioner_3 = pipeline(model="ford442/blip-image-to-text-large-bf16",device='cuda', task="image-to-text")
+model5 = Blip2ForConditionalGeneration.from_pretrained("ford442/blip2-image-to-text-bf16").to('cuda')
+processor5 = Blip2Processor.from_pretrained("ford442/blip2-image-to-text-bf16")
 txt_tokenizer = AutoTokenizer.from_pretrained(checkpoint, add_prefix_space=False)
 txt_tokenizer.tokenizer_legacy=False
 model = AutoModelForCausalLM.from_pretrained(checkpoint).to('cuda')