Spaces:

flax-community
/

koclip

Build error

App Files Files Community

koclip / app.py

jaketae's picture

style: run linter

2cf3514 almost 3 years ago

raw history blame

No virus

329 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)