mirror of
https://github.com/csd4ni3l/grass_touching_captcha.git
synced 2026-01-01 04:23:45 +01:00
fix model again and fix images being inputted as text
This commit is contained in:
5
jina.py
5
jina.py
@@ -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)
|
||||||
|
|||||||
Reference in New Issue
Block a user