Update model name

app.py

@@ -7,7 +7,7 @@ import text2image
 PAGES = {"Text to Image": text2image, "Image to Text": image2text}
 
 st.sidebar.title("Navigation")
-model = st.sidebar.selectbox("Choose a model", ["koclip", "koclip-large"])
+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)

embed.py

@@ -16,7 +16,7 @@ def main(args):
     files = list(os.listdir(root))
     for f in files:
         assert(f[-4:] == ".jpg")
-    for model_name in ["koclip", "koclip-large"]:
+    for model_name in ["koclip-base", "koclip-large"]:
         model, processor = load_model(f"koclip/{model_name}")
         with tqdm(total=len(files)) as pbar:
             for counter in range(0, len(files), args.batch_size):