image2text.py

import streamlit as st

from utils import load_model


def app(model_name):
    model, processor = load_model(model_name)

    st.title("Text to Image Retrieval")
    st.markdown("""
        Some text goes in here.
    """)