Update app.py

app.py

@@ -4,7 +4,7 @@ import torch
 from transformers import Blip2Processor, Blip2ForConditionalGeneration
 
 
-def load_model():
+def load_caption_model():
 
     processor = Blip2Processor.from_pretrained("Salesforce/blip2-opt-2.7b")
     model = Blip2ForConditionalGeneration.from_pretrained("Salesforce/blip2-opt-2.7b", load_in_8bit=True,torch_dtype=torch.float16, device_map="auto")