computer-vision-backend

Runtime error

Abhilashvj commited on Sep 18, 2023

Commit

6514fe3

1 Parent(s): 8b3bdd0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -218,7 +218,7 @@ async def image_search_pinecone(
             print("Indexing query image...")
             query_image = enhance_image(query_image)
             if model_type =="huggingface":
-                prompt_embedding = get_clip_vectors(query_image)
             else:
                 prompt_embedding = image_sim_model.encode(query_image, convert_to_tensor=True).tolist()
             if INDEX_NAME not in pinecone.list_indexes():

             print("Indexing query image...")
             query_image = enhance_image(query_image)
             if model_type =="huggingface":
+                prompt_embedding = get_clip_vectors(query_image).tolist()
             else:
                 prompt_embedding = image_sim_model.encode(query_image, convert_to_tensor=True).tolist()
             if INDEX_NAME not in pinecone.list_indexes():