File size: 388 Bytes
bef8931
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
from __main__ import initalize_models, load_projection_module, projection_module

tokenizer, model, vision_model, processor = initialize_models()
projection_module = load_projection_module()

img = st.file_uploader("image: ")
text = st.text_area("text: ")

ans = answer_question(
    img,
    tokenizer,
    model,
    vision_model,
    processor,
    projection_module,
)

st.json(ans)