Update app.py

app.py

@@ -78,7 +78,7 @@ def load_model(model_selection):
 # Choose LLM to use
 # weights for OPT-6.7B/OPT-13B/OPT-30B/OPT-66B will download automatically
 print("Loading Large Language Model (LLM)...")
-llm_model, tokenizer = load_model('facebook/opt-1.3b')  # ~13G (FP16)
+llm_model, tokenizer = load_model('facebook/opt-350m')  # ~13G (FP16)
 llm_model.to(device)
 model, vis_processors, txt_processors = load_model_and_preprocess(name="img2prompt_vqa", model_type="base", is_eval=True, device=device)