Update model name

utils.py

@@ -22,8 +22,8 @@ def load_index(img_file):
     return filenames, index
 
 @st.cache(allow_output_mutation=True)
-def load_model(model_name="koclip/koclip"):
-    assert model_name in {"koclip/koclip", "koclip/koclip-large"}
+def load_model(model_name="koclip/koclip-base"):
+    assert model_name in {"koclip/koclip-base", "koclip/koclip-large"}
     model = FlaxHybridCLIP.from_pretrained(model_name)
     processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch32")
     processor.tokenizer = AutoTokenizer.from_pretrained("klue/roberta-large")