koclip / app.py
ampehta
Revert "Merge branch 'main' of https://huggingface.co/spaces/flax-community/koclip into main"
503acf7
raw history blame
No virus
330 Bytes
import streamlit as st
import image2text
import text2image
PAGES = {"Text to Image": text2image, "Image to Text": image2text}
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)