switching to pretrain_flant5xl

app.py

@@ -9,7 +9,7 @@ from PIL import Image
 device = torch.device("cuda") if torch.cuda.is_available() else "cpu"
 
 model, vis_processors, _ = load_model_and_preprocess(
-    name="blip2_opt", model_type="pretrain_opt2.7b", is_eval=True, device=device
+    name="blip2_t5", model_type="pretrain_flant5xl", is_eval=True, device=device
 )
 
 
@@ -62,7 +62,7 @@ with gr.Blocks() as demo:
         "### BLIP-2: Bootstrapping Language-Image Pre-training with Frozen Image Encoders and Large Language Models"
     )
     gr.Markdown(
-        "This demo uses `OPT2.7B` weights. For more information please see [Github](https://github.com/salesforce/LAVIS/tree/main/projects/blip2) or [Paper](https://arxiv.org/abs/2301.12597)."
+        "This demo uses the `pretrain_flant5xl` weights. For more information please visit [Github](https://github.com/salesforce/LAVIS/tree/main/projects/blip2) or [Paper](https://arxiv.org/abs/2301.12597)."
     )
 
     with gr.Row():