Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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-
|
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.
|
106 |
-
<h4> 3.
|
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=
|