vumichien commited on
Commit
ec0930e
1 Parent(s): c7e5b16

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -78,7 +78,7 @@ def load_model(model_selection):
78
  # Choose LLM to use
79
  # weights for OPT-6.7B/OPT-13B/OPT-30B/OPT-66B will download automatically
80
  print("Loading Large Language Model (LLM)...")
81
- llm_model, tokenizer = load_model('facebook/opt-350m') # ~13G (FP16)
82
  llm_model.to(device)
83
  model, vis_processors, txt_processors = load_model_and_preprocess(name="img2prompt_vqa", model_type="base", is_eval=True, device=device)
84
 
@@ -102,9 +102,8 @@ with demo:
102
  <div align="center">
103
  <h3> What you can do with this space </h3>
104
  <h4> 1. Upload your image and fill your question </h4>
105
- <h4> 2. Generating gradcam attention from model </h4>
106
- <h4> 3. Creating caption from your image </h4>
107
- <h4> 4. Answering your question based on uploaded image </h4>
108
  </div>
109
  ''')
110
  examples = gr.Examples(examples=
 
78
  # Choose LLM to use
79
  # weights for OPT-6.7B/OPT-13B/OPT-30B/OPT-66B will download automatically
80
  print("Loading Large Language Model (LLM)...")
81
+ llm_model, tokenizer = load_model('facebook/opt-1.3b') # ~13G (FP16)
82
  llm_model.to(device)
83
  model, vis_processors, txt_processors = load_model_and_preprocess(name="img2prompt_vqa", model_type="base", is_eval=True, device=device)
84
 
 
102
  <div align="center">
103
  <h3> What you can do with this space </h3>
104
  <h4> 1. Upload your image and fill your question </h4>
105
+ <h4> 2. Creating caption from your image </h4>
106
+ <h4> 3. Answering your question based on uploaded image </h4>
 
107
  </div>
108
  ''')
109
  examples = gr.Examples(examples=