Update app.py

app.py

@@ -76,9 +76,9 @@ def load_model(model_selection):
     return model,tokenizer
 
 # Choose LLM to use
-# weights for OPT-6.7B/OPT-13B/OPT-30B/OPT-66B will download automatically
+# weights for OPT-350M/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')  # ~700MB (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)