Update app.py

app.py

@@ -9,7 +9,7 @@ model = hub.load(model_url)
 
 def embed_text(text: str) -> dict:
     embeddings = model(text)
-    return embeddings.numpy().tolist()
+    return embeddings.numpy().tolist()[0]
 
 embed_text_inter = gr.Interface(
     fn = embed_text,
@@ -19,11 +19,16 @@ embed_text_inter = gr.Interface(
 )
 
 def distance(text_1: str, text_2: str) -> float:
-    embeddings_1 = model(text_1)
-    embeddings_2 = model(text_2)
-    dist = math.sqrt(sum((embeddings_1 - embeddings_2)**2))
+    embeddings_1 = embed_text(text_1)
+    embeddings_2 = embed_text(text_2)
+    
+    dist = 0
+    for i in range(len(embeddings_1)):
+        dist += (embeddings_1[i] - embeddings_2[i]) ** 2
+    dist = math.sqrt(dist)
     return dist
 
+
 distance_inter = gr.Interface(
     fn = distance,
     inputs = ["text", "text"],