image2text.py

import streamlit as st

from utils import load_model


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

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