neiths's picture
Create app.py
bef8931 verified
raw
history blame contribute delete
388 Bytes
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)