import streamlit as st

import image2text
import text2image


PAGES = {"Text to Image": text2image, "Image to Text": image2text}

st.sidebar.title("Navigation")
model = st.sidebar.radio("Model", ["koclip/koclip", "koclip/koclip-large"])
page = st.sidebar.radio("Go to", list(PAGES.keys()))

PAGES[page].app(model)