shivangibithel commited on
Commit
7320ee9
1 Parent(s): b68c4cb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -51,7 +51,7 @@ zip_file = zipfile.ZipFile(zip_path)
51
 
52
  text_model = CLIPTextModelWithProjection.from_pretrained("openai/clip-vit-base-patch32")
53
  text_tokenizer = AutoTokenizer.from_pretrained("openai/clip-vit-base-patch32")
54
- d = 1024
55
  text_index = faiss.index_factory(d, "Flat", faiss.METRIC_INNER_PRODUCT)
56
  text_index = faiss.read_index("text_index.index")
57
 
 
51
 
52
  text_model = CLIPTextModelWithProjection.from_pretrained("openai/clip-vit-base-patch32")
53
  text_tokenizer = AutoTokenizer.from_pretrained("openai/clip-vit-base-patch32")
54
+ d = 32
55
  text_index = faiss.index_factory(d, "Flat", faiss.METRIC_INNER_PRODUCT)
56
  text_index = faiss.read_index("text_index.index")
57