feature: change radio buttons to dropdown

app.py

@@ -7,7 +7,7 @@ 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()))
+model = st.sidebar.selectbox("Choose a model", ["koclip", "koclip-large"])
+page = st.sidebar.selectbox("Choose a task", list(PAGES.keys()))
 
-PAGES[page].app(model)
+PAGES[page].app(f"koclip/{model}")

image2text.py

@@ -6,7 +6,7 @@ from utils import load_model
 def app(model_name):
     model, processor = load_model(model_name)
 
-    st.title("Text to Image Retrieval")
+    st.title("Image to Text")
     st.markdown("""
         Some text goes in here.
     """)

text2image.py

@@ -7,7 +7,7 @@ def app(model_name):
     model, processor = load_model(model_name)
 
 
-    st.title("Text to Image Retrieval")
+    st.title("Text to Image")
     st.markdown("""
         Some text goes in here.
     """)