File size: 421 Bytes
f1d50b1
 
 
84c806e
14261e1
f1d50b1
14261e1
 
 
 
 
f1d50b1
 
532be59
b1f513f
f1d50b1
cf349fd
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
import streamlit as st

import image2text
import most_relevant_part
import text2image

PAGES = {
    "Text to Image": text2image,
    "Image to Text": image2text,
    "Most Relevant Part of Image": most_relevant_part,
}

st.sidebar.title("Navigation")
model = st.sidebar.selectbox("Choose a model", ["koclip-base", "koclip-large"])
page = st.sidebar.selectbox("Choose a task", list(PAGES.keys()))

PAGES[page].app(model)