Update app.py

app.py

@@ -6,11 +6,17 @@ from pinecone import Pinecone
 from sentence_transformers import SentenceTransformer
 from datasets import load_dataset
 import os
-import re
+import 
+####################
+import pandas as pd
+##########################
 
 model = SentenceTransformer('sentence-transformers/clip-ViT-B-32')
 
-fashion = load_dataset("ashraq/fashion-product-images-small", split="train")
+fashion = load_dataset("ashraq/fashion-product-images-small", split="train
+###############
+fashion_df = pd.DataFrame(fashion)
+####################
 images = fashion['image']
 metadata = fashion.remove_columns('image')
 item_list = list(set(metadata['productDisplayName']))
@@ -66,7 +72,8 @@ def process_input(query, slider_value):
     search_words = query.title().split()
     pattern = r"(?=.*\b" + r"\b)(?=.*\b".join(map(re.escape, search_words)) + r"\b)"
     filtered_items = [item for item in item_list if re.search(pattern, item)]
-    filtered_df = fashion.filter(lambda x: x['productDisplayName'] in filtered_items)
+    # filtered_df = fashion.filter(lambda x: x['productDisplayName'] in filtered_items)
+    filtered_df = fashion_df[fashion_df['productDisplayName'].isin(filtered_items)]
     #####################
     try:
         slider_value = float(slider_value)
@@ -85,8 +92,11 @@ def process_input(query, slider_value):
         matches = [x["metadata"]['productDisplayName'] for x in result["matches"]]
 
         ###########
-        exact_match = filtered_df.filter(lambda x: x['productDisplayName'] == query)[0]['productDisplayName']
-        exact_img = filtered_df.filter(lambda x: x['productDisplayName'] == query)[0]['image']
+        # exact_match = filtered_df.filter(lambda x: x['productDisplayName'] == query)[0]['productDisplayName']
+        # exact_img = filtered_df.filter(lambda x: x['productDisplayName'] == query)[0]['image']
+        exact_match = filtered_df[filtered_df['productDisplayName']==query]['productDisplayName']
+        exact_img = filtered_df[filtered_df['productDisplayName']==query]['image']
+
         if exact_match is not None:
             imgs.insert(0, exact_img)
             matches.insert(0, exact_match)