Update app.py

app.py

@@ -113,7 +113,7 @@ def detect_clothing(image):
     for detection in detections:
         x1, y1, x2, y2, conf, cls = detection
         category = yolo_model.names[int(cls)]
-        if category in ['top', 'bottom', 'hat', 'shoes']:  # Add more categories as needed
+        if category in ['sunglass','hat','jacket','shirt','pants','shorts','skirt','dress','bag','shoe']:
             categories.append({
                 'category': category,
                 'bbox': [int(x1), int(y1), int(x2), int(y2)],
@@ -146,6 +146,7 @@ if search_type == "Image URL":
                     if st.button("Search Similar Items"):
                         selected_detection = next(d for d in detections if f"{d['category']} (Confidence: {d['confidence']:.2f})" == selected_category)
                         cropped_image = crop_image(query_image, selected_detection['bbox'])
+                        st.image(cropped_image, caption="Cropped Image", use_column_width=True)
                         query_embedding = get_image_embedding(cropped_image)
                         similar_images = find_similar_images(query_embedding)