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)