fix model again and fix images being inputted as text

This commit is contained in:
csd4ni3l
2025-08-08 23:14:45 +02:00
parent 881ccbda13
commit 3f304071e3

View File

@@ -14,7 +14,8 @@ def get_embedding(input: list):
headers["Authorization"] = f"Bearer {os.environ['JINA_TOKEN']}" headers["Authorization"] = f"Bearer {os.environ['JINA_TOKEN']}"
data = { data = {
"model": "jina-clip-v2", "model": "jina-embeddings-v4",
"task": "text-matching",
"input": input "input": input
} }
@@ -23,7 +24,7 @@ def get_embedding(input: list):
return [jina_object["embedding"] for jina_object in response.json()["data"]] return [jina_object["embedding"] for jina_object in response.json()["data"]]
def get_grass_touching_similarity(image_url): def get_grass_touching_similarity(image_url):
grass_image_embedding, grass_word_embedding = get_embedding([image_url, "hand touching grass"]) grass_image_embedding, grass_word_embedding = get_embedding([{"image": image_url}, {"text": "hand touching grass"}])
grass_image_embedding = np.array(grass_image_embedding) grass_image_embedding = np.array(grass_image_embedding)
grass_word_embedding = np.array(grass_word_embedding) grass_word_embedding = np.array(grass_word_embedding)