vsrinivas commited on
Commit
773c328
·
verified ·
1 Parent(s): 86e3e6d

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +21 -0
app.py ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from io import BytesIO
3
+ from base64 import b64encode
4
+ from pinecone_text.sparse import BM25Encoder
5
+ from pinecone import Pinecone
6
+ from sentence_transformers import SentenceTransformer
7
+ from datasets import load_dataset
8
+
9
+ model = SentenceTransformer('sentence-transformers/clip-ViT-B-32')
10
+
11
+ fashion = load_dataset("ashraq/fashion-product-images-small", split="train")
12
+ images = fashion['image']
13
+ metadata = fashion.remove_columns('image')
14
+
15
+ INDEX_NAME = 'srinivas-hybrid-search'
16
+ PINECONE_API_KEY = os.getenv(pinecone_api_key)
17
+ pinecone = Pinecone(api_key=PINECONE_API_KEY)
18
+ index = pinecone.Index(INDEX_NAME)
19
+ bm25 = BM25Encoder()
20
+ bm25.fit(metadata['productDisplayName'])
21
+