ad

README.md

@@ -1,7 +1,7 @@
 ---
-title: Clip Distance
-emoji: 📈
-colorFrom: gray
+title: CLIP Distance
+emoji: 😻
+colorFrom: blue
 colorTo: indigo
 sdk: gradio
 sdk_version: 4.22.0

app.py

@@ -0,0 +1,15 @@
+import gradio as gr
+from sentence_transformers import SentenceTransformer
+from PIL import Image
+from scipy.spatial import distance
+
+model = SentenceTransformer('clip-ViT-B-32')
+
+def image_distance(image1, image2) -> float:
+    embedding1 = model.encode(sentences=[Image.fromarray(image1)], batch_size=128)
+    embedding2 = model.encode(sentences=[Image.fromarray(image2)], batch_size=128)
+
+    return 1 - distance.cosine(embedding1, embedding2)
+
+demo = gr.Interface(fn=image_distance, inputs=["image", "image"], outputs="label")
+demo.launch()

requirements.txt

@@ -0,0 +1,4 @@
+gradio~=3.41.2
+sentence-transformers
+numpy~=1.25.2
+scipy