Spaces:

shivangibithel
/

LCMI_T2I

Running

shivangibithel commited on Mar 15, 2023

Commit

1bd0451

•

1 Parent(s): 8d613df

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -43,8 +43,8 @@ zip_file = zipfile.ZipFile(zip_path)
 # text_model = CLIPTextModelWithProjection.from_pretrained("openai/clip-vit-base-patch32")
 # text_tokenizer = AutoTokenizer.from_pretrained("openai/clip-vit-base-patch32")
-text_index = faiss.read_index("text_index.index")
 def T2Isearch(query, k=50):
     # Encode the text query

 # text_model = CLIPTextModelWithProjection.from_pretrained("openai/clip-vit-base-patch32")
 # text_tokenizer = AutoTokenizer.from_pretrained("openai/clip-vit-base-patch32")
+text_index = faiss.index_factory(vectors.shape[1], "Flat", faiss.METRIC_INNER_PRODUCT)
+faiss.read_index("text_index.index")
 def T2Isearch(query, k=50):
     # Encode the text query