fix: rename, etc

app.py

@@ -7,8 +7,10 @@ import torch
 from sentence_transformers.util import semantic_search
 from transformers import AutoModel, AutoProcessor
 
-st.title("VitB32 Bert Ko Small Clip Test")
-st.markdown("Unsplash data에서 입력 텍스트와 가장 유사한 이미지를 검색합니다.")
+st.title("My CLIP Model Test")
+st.markdown(
+    "[Unsplash Lite dataset](https://unsplash.com/data)에서 입력 텍스트와 가장 유사한 이미지를 검색합니다."
+)
 
 
 @st.cache(allow_output_mutation=True, show_spinner=False)
@@ -34,8 +36,10 @@ model_type = st.radio(
 
 if model_type == "Bingsu/clip-vit-base-patch32-ko":
     model, processor = get_clip_model()
-else:
+elif model_type == "Bingsu/vitB32_bert_ko_small_clip":
     model, processor = get_dual_encoder_model()
+else:
+    raise ValueError("Invalid model type")
 
 info = pd.read_csv("info.csv")
 with open("img_id.pkl", "rb") as f: