Update app.py

app.py

@@ -12,17 +12,23 @@ from ultralytics import YOLO
 import cv2
 import chromadb
 
-# CLIP 모델 로드
 @st.cache_resource
 def load_clip_model():
     model, preprocess_train, preprocess_val = open_clip.create_model_and_transforms('hf-hub:Marqo/marqo-fashionSigLIP')
     tokenizer = open_clip.get_tokenizer('hf-hub:Marqo/marqo-fashionSigLIP')
+    
+    # 파인튜닝한 모델의 state_dict 불러오기
+    state_dict = torch.load('./accessory_clip.pt', map_location=torch.device('cpu'))
+    model.load_state_dict(state_dict)  # 모델에 state_dict 적용
+    
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     model.to(device)
+    
     return model, preprocess_val, tokenizer, device
 
 clip_model, preprocess_val, tokenizer, device = load_clip_model()
 
+
 @st.cache_resource
 def load_yolo_model():
     return YOLO("./accessaries.pt")
@@ -116,12 +122,12 @@ if 'selected_category' not in st.session_state:
     st.session_state.selected_category = None
 
 # Streamlit app
-st.title("Accessory Search App")
+st.title("Accessary Search App")
 
 # 단계별 처리
 if st.session_state.step == 'input':
     st.session_state.query_image_url = st.text_input("Enter image URL:", st.session_state.query_image_url)
-    if st.button("Detect accessory"):
+    if st.button("Detect accesseary"):
         if st.session_state.query_image_url:
             query_image = load_image_from_url(st.session_state.query_image_url)
             if query_image is not None: